Skip to content

Commit bc121b9

Browse files
committed
update
1 parent be1dabc commit bc121b9

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

  • fastdeploy/model_executor/layers/moe

fastdeploy/model_executor/layers/moe/moe.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ def get_moe_method():
3838
return moe method based on device platform
3939
"""
4040

41-
if current_platform.is_cuda():
41+
if current_platform.is_cuda() or current_platform.is_iluvatar():
4242
from .fused_moe_cutlass_backend import CutlassMoEMethod
4343

4444
return CutlassMoEMethod(None)

0 commit comments

Comments
 (0)