hotfix: rename moe/test_utils.py to moe/utils.py (#2106)

yzh119 · web-flow · commit 1c4b52232434 · 2025-11-18T12:53:43.000-08:00
diff --git a/tests/moe/test_dpsk_fused_moe_fp8.py b/tests/moe/test_dpsk_fused_moe_fp8.py
@@ -7,7 +7,7 @@
     WeightLayout,
     trtllm_fp8_block_scale_moe,
 )
-from .test_utils import skip_checks, QuantMode
+from .utils import skip_checks, QuantMode
 from flashinfer import GatedActType
 
 
@@ -611,7 +611,7 @@ def __init__(self):
     if "compatible_moe_impls" not in weight_processing:
         weight_processing["compatible_moe_impls"] = [type(moe_impl)]
 
-    # Use the complete skip_checks function from test_utils
+    # Use the complete skip_checks function from utils
     skip_checks(
         moe_impl=moe_impl,
         routing_config=routing_config,
diff --git a/tests/moe/test_trtllm_gen_fused_moe.py b/tests/moe/test_trtllm_gen_fused_moe.py
@@ -45,7 +45,7 @@
     get_w2_permute_indices_with_cache,
     _maybe_get_cached_w3_w1_permute_indices,
 )
-from .test_utils import skip_checks, QuantMode
+from .utils import skip_checks, QuantMode
 
 
 # Max num tokens to tune for trtllm-gen fused moe
diff --git a/tests/moe/utils.py b/tests/moe/utils.py

Original file line number	Diff line number	Diff line change
`@@ -45,7 +45,7 @@`
`45`	`45`	`get_w2_permute_indices_with_cache,`
`46`	`46`	`_maybe_get_cached_w3_w1_permute_indices,`
`47`	`47`	`)`
`48`		`-from .test_utils import skip_checks, QuantMode`
	`48`	`+from .utils import skip_checks, QuantMode`
`49`	`49`
`50`	`50`
`51`	`51`	`# Max num tokens to tune for trtllm-gen fused moe`