Fix torch.compile for MoE (#2033)

2024-11-14 01:30:24 -08:00
parent b275ce0043
commit c3eac1b010
10 changed files with 89 additions and 12 deletions
--- a/test/srt/run_suite.py
+++ b/test/srt/run_suite.py
@@ -27,6 +27,7 @@ suites = {
        "test_srt_engine.py",
        "test_srt_endpoint.py",
        "test_torch_compile.py",
+        "test_torch_compile_moe.py",
        "test_torchao.py",
        "test_triton_attention_kernels.py",
        "test_triton_attention_backend.py",