[1/n] Fix hanging during DeepGemm Warmup (#14493)

This commit is contained in:
Baizhou Zhang
2025-12-06 10:44:02 -08:00
committed by GitHub
parent 3e40c63674
commit bc388471d2
2 changed files with 49 additions and 12 deletions
-10
View File
@@ -1,4 +1,3 @@
import os
import unittest
from types import SimpleNamespace
@@ -49,10 +48,6 @@ class TestDeepseek(CustomTestCase):
"2048",
"--disable-radix-cache",
],
env={
**os.environ,
"SGLANG_JIT_DEEPGEMM_PRECOMPILE": "0",
},
)
@classmethod
@@ -75,7 +70,6 @@ class TestDeepseek(CustomTestCase):
self.assertGreater(metrics["accuracy"], 0.92)
@unittest.skip("Can pass locally, but will cause Timeout on CI runner.")
class TestDeepseekMTP(CustomTestCase):
@classmethod
def setUpClass(cls):
@@ -118,10 +112,6 @@ class TestDeepseekMTP(CustomTestCase):
"2",
"--disable-radix-cache",
],
env={
**os.environ,
"SGLANG_JIT_DEEPGEMM_PRECOMPILE": "0",
},
)
@classmethod