Skip to content

Commit f3dae20

Browse files
committed
Fix flashinfer/fused_moe.__init__.py import issues found by 'pre-commit run --all-files'
Signed-off-by: amitz-nv <203509407+amitz-nv@users.noreply.github.com>
1 parent 6d9ed8a commit f3dae20

1 file changed

Lines changed: 7 additions & 4 deletions

File tree

flashinfer/fused_moe/__init__.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -15,10 +15,6 @@
1515
"""
1616

1717
from .core import (
18-
ActivationType,
19-
Fp8QuantizationType,
20-
RoutingMethodType,
21-
WeightLayout,
2218
convert_to_block_layout,
2319
cutlass_fused_moe,
2420
gen_cutlass_fused_moe_sm120_module,
@@ -37,6 +33,13 @@
3733
trtllm_mxint4_block_scale_moe,
3834
)
3935

36+
from ..tllm_enums import (
37+
ActivationType,
38+
Fp8QuantizationType,
39+
WeightLayout,
40+
RoutingMethodType,
41+
)
42+
4043
from .fused_routing_dsv3 import ( # noqa: F401
4144
fused_topk_deepseek as fused_topk_deepseek,
4245
)

0 commit comments

Comments
 (0)