debug: remove assert, test token corruption

This commit is contained in:
2026-05-17 08:19:45 +00:00
parent 66627926c5
commit 2bbe04efd8

View File

@@ -86,12 +86,6 @@ class CuTeDSLMoERunner:
self.max_num_tokens, device=self.device, dtype=torch.int32
).unsqueeze(1).expand(-1, self.top_k).contiguous().view(-1)
# Force GPU sync and verify
torch.cuda.synchronize()
_check = self._token_indices[:8].cpu().tolist()
assert _check == [0, 0, 1, 1, 2, 2, 3, 3], \
f"Token indices corrupted: {_check}"
self._expert_id_range = torch.arange(self.num_experts, device=self.device)
self._expert_offsets_buf = torch.zeros(