[Iluvatar][CI] fix safetensors_rust.SafetensorError: framework paddle is invalid (#4972)

2026-04-23 00:17:25 +08:00 · 2025-11-12 14:13:40 +08:00
parent 35bd2afab3
commit 76e60e98f8
3 changed files with 3 additions and 2 deletions
@@ -38,7 +38,7 @@ def get_moe_method():
    return moe method based on device platform
    """

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

        return CutlassMoEMethod(None)