[Clean code] Compressed_tensors_moe code clean (#13719)

This commit is contained in:
Xiaoyu Zhang
2025-11-21 18:15:46 +08:00
committed by GitHub
parent 6d0e0b9bfc
commit a34d3abb54

View File

@@ -52,11 +52,6 @@ if _use_aiter:
logger = logging.getLogger(__name__)
def _mask_topk_ids_cpu_experts(topk_ids: torch.Tensor, num_gpu_experts: int):
"""Mask topk_ids >= num_gpu_experts by setting them to -1."""
topk_ids[topk_ids >= num_gpu_experts] = -1
class GPTQMarlinState(Enum):
REPACK = enum.auto()
READY = enum.auto()