From 714dd1a4a980f7937a74343d19a8eba4fe321480 Mon Sep 17 00:00:00 2001 From: Chenggang Zhao Date: Mon, 11 May 2026 19:20:18 +0800 Subject: [PATCH] Update test_mega_moe.py --- tests/test_mega_moe.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_mega_moe.py b/tests/test_mega_moe.py index e74b65e..83e8d62 100644 --- a/tests/test_mega_moe.py +++ b/tests/test_mega_moe.py @@ -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():