Update test_mega_moe.py

2026-05-11 19:20:18 +08:00
parent 891d57b4db
commit 714dd1a4a9
1 changed files with 1 additions and 1 deletions
@@ -151,7 +151,7 @@ def test(local_rank: int, num_local_ranks: int, args: argparse.Namespace):
        num_topk=num_topk, use_fp8_dispatch=True,
        explicitly_destroy=True,
        allow_multiple_reduction=False,
-        gpu_timeout_secs=10, cpu_timeout_secs=30
+        num_gpu_timeout_secs=10, num_cpu_timeout_secs=30
    ) if is_legacy_loaded else None
    def run_baseline():