Bump sgl-deep-gemm to 0.1.7 (#37279)
This commit is contained in:
@@ -115,7 +115,7 @@ class TestPrepareServerArgs(CustomTestCase):
|
||||
# daemon to build the same static EPLB layout as the engine.
|
||||
handle_load_format(args)
|
||||
|
||||
def test_enable_w4a4_mxfp4_megamoe_sets_deepgemm_env(self):
|
||||
def test_enable_w4a4_mxfp4_megamoe_preserves_legacy_deepgemm_env(self):
|
||||
deepgemm_env = {
|
||||
"DG_USE_FP4_ACTS": "0",
|
||||
"DG_USE_MXF4_KIND": "0",
|
||||
@@ -134,8 +134,8 @@ class TestPrepareServerArgs(CustomTestCase):
|
||||
args.resolve_once()
|
||||
|
||||
self.assertTrue(resolution_result(args, "enable_w4a4_mxfp4_megamoe"))
|
||||
self.assertEqual(os.environ["DG_USE_FP4_ACTS"], "1")
|
||||
self.assertEqual(os.environ["DG_USE_MXF4_KIND"], "1")
|
||||
self.assertEqual(os.environ["DG_USE_FP4_ACTS"], "0")
|
||||
self.assertEqual(os.environ["DG_USE_MXF4_KIND"], "0")
|
||||
|
||||
def test_w4a4_mxfp4_megamoe_disabled_preserves_deepgemm_env(self):
|
||||
deepgemm_env = {
|
||||
|
||||
Reference in New Issue
Block a user