diff --git a/test/srt/ep/test_moe_ep.py b/test/srt/ep/test_moe_ep.py index 74a5790d4..bccd2673a 100644 --- a/test/srt/ep/test_moe_ep.py +++ b/test/srt/ep/test_moe_ep.py @@ -66,6 +66,8 @@ class TestEpDeepGEMM(CustomTestCase): "fp8", "--moe-runner-backend", "deep_gemm", + "--moe-a2a-backend", + "deepep", ], ) diff --git a/test/srt/run_suite.py b/test/srt/run_suite.py index 13f880d53..6574ac8a6 100644 --- a/test/srt/run_suite.py +++ b/test/srt/run_suite.py @@ -135,7 +135,6 @@ suites = { TestFile("test_modelopt_export.py", 30), ], "per-commit-2-gpu": [ - TestFile("ep/test_moe_ep.py", 140), TestFile("hicache/test_hicache_storage_3fs_backend.py", 200), TestFile("hicache/test_hicache_storage_file_backend.py", 200), TestFile("hicache/test_hicache_storage_mooncake_backend.py", 300), @@ -202,6 +201,9 @@ suites = { ], # Nightly test suites have been moved to test/run_suite_nightly.py "__not_in_ci__": [ + TestFile( + "ep/test_moe_ep.py", 140 + ), # Temporarily disabled, need to fix ep errors TestFile("test_release_memory_occupation.py", 200), # Temporarily disabled TestFile("models/test_dummy_grok_models.py"), TestFile(