flashinfer-ai
diff --git a/‎flashinfer/comm/trtllm_moe_a2a.py‎ renamed to ‎flashinfer/comm/trtllm_moe_alltoall.py‎ b/‎flashinfer/comm/trtllm_moe_a2a.py‎ renamed to ‎flashinfer/comm/trtllm_moe_alltoall.py‎
diff --git a/‎scripts/task_test_multi_node_comm_kernels.sh‎
Lines changed: 1 addition & 1 deletion b/‎scripts/task_test_multi_node_comm_kernels.sh‎
Lines changed: 1 addition & 1 deletion
diff --git a/‎tests/comm/test_mnnvl_a2a.py‎ renamed to ‎tests/comm/test_mnnvl_moe_alltoall.py‎ b/‎tests/comm/test_mnnvl_a2a.py‎ renamed to ‎tests/comm/test_mnnvl_moe_alltoall.py‎
@@ -16,4 +16,4 @@ set -x
 
 pytest -s tests/comm/test_mnnvl_memory.py
 pytest -s tests/comm/test_trtllm_mnnvl_allreduce.py
-pytest -s tests/comm/test_mnnvl_a2a.py
+pytest -s tests/comm/test_mnnvl_moe_alltoall.py