Skip to content

Commit 67c5113

Browse files
committed
Add e2e test for multistream moe
Signed-off-by: sdmyzlp <lrwei2@petalmail.com>
1 parent 7859887 commit 67c5113

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

tests/e2e/multicard/test_torchair_graph_mode.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -99,3 +99,19 @@ def test_e2e_deepseekv3_with_torchair_ms_mla():
9999
},
100100
}
101101
_deepseek_torchair_test_fixture(additional_config)
102+
103+
104+
@pytest.mark.skipif(os.getenv("VLLM_USE_V1") == "0",
105+
reason="torchair graph is not supported on v0")
106+
def test_e2e_deepseekv3_with_torchair_ms_moe():
107+
additional_config = {
108+
"torchair_graph_config": {
109+
"enabled": True,
110+
"enable_multistream_moe": True,
111+
},
112+
"ascend_scheduler_config": {
113+
"enabled": True,
114+
},
115+
"refresh": True,
116+
}
117+
_deepseek_torchair_test_fixture(additional_config)

0 commit comments

Comments
 (0)