remove unused var

Signed-off-by: CaranLic <[email protected]>

Author: Liccol

tests/e2e/singlecard/ops/test_fused_moe.py

@@ -118,12 +118,6 @@ def test_token_dispatcher_with_all_gather(
     score = torch.softmax(score, dim=-1, dtype=dtype)
     topk_weights, topk_ids = torch.topk(score, topk)
     topk_ids = topk_ids.to(torch.int32)
-    row_idx = (torch.arange(
-        0,
-        m * topk,
-        device=device,
-        dtype=torch.int32,
-    ).view(topk, -1).permute(1, 0).contiguous())
 
     dispatcher_kwargs = {
         "num_experts": e,
@@ -200,12 +194,6 @@ def test_token_dispatcher_with_all_gather_quant(
         score = torch.softmax(score, dim=-1, dtype=dtype)
         topk_weights, topk_ids = torch.topk(score, topk)
         topk_ids = topk_ids.to(torch.int32)
-        row_idx = (torch.arange(
-            0,
-            m * topk,
-            device=device,
-            dtype=torch.int32,
-        ).view(topk, -1).permute(1, 0).contiguous())
 
         dispatcher_kwargs = {
             "num_experts": e,

tests/ut/ops/test_moe_comm_method.py

@@ -204,7 +204,6 @@ def test_fused_experts_method(self, mock_unified_apply_mlp,
         topk_weights = torch.tensor([[0.5, 0.5], [0.3, 0.7], [0.8, 0.2],
                                      [0.6, 0.4]])
         topk_ids = torch.tensor([[0, 1], [1, 2], [2, 0], [1, 1]])
-        row_idx = torch.arange(4)
 
         # Make sure tensors are contiguous and have correct strides
         hidden_states = hidden_states.contiguous()