Skip to content

Commit 31c4c80

Browse files
committed
fixes
Signed-off-by: Tyler Michael Smith <tyler@neuralmagic.com>
1 parent 1a31242 commit 31c4c80

File tree

1 file changed

+2
-1
lines changed
  • vllm/model_executor/layers/fused_moe

1 file changed

+2
-1
lines changed

vllm/model_executor/layers/fused_moe/layer.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,8 @@
2424
from vllm.utils import direct_register_custom_op
2525

2626
if current_platform.is_cuda_alike():
27-
from .fused_moe import cutlass_moe, fused_experts
27+
from .cutlass_moe import cutlass_moe
28+
from .fused_moe import fused_experts
2829
else:
2930
fused_experts = None # type: ignore
3031
cutlass_moe = None # type: ignore

0 commit comments

Comments
 (0)