Re-introduce the unit test of test_mooncake_ep_small (#16019)

2026-01-12 17:01:24 +08:00
parent b1ee75ae7b
commit 9f5cd80a8d
4 changed files with 17 additions and 21 deletions
@@ -45,8 +45,7 @@ suites = {
    ],
    "per-commit-4-gpu-deepep": [
        TestFile("ep/test_deepep_small.py", 531),
-        # TODO: Add it back after mooncake supports torch 2.9
-        # TestFile("ep/test_mooncake_ep_small.py", 450),
+        TestFile("ep/test_mooncake_ep_small.py", 660),
    ],
    "per-commit-8-gpu-h200-deepep": [
        TestFile("ep/test_deepep_large.py", 563),