[MUSA][Diffusion] Improve wan model inference speed using torch.compile (#25256)

Co-authored-by: R0CKSTAR <yeahdongcn@gmail.com>
2026-05-17 22:10:24 +08:00
parent eccfd6dea7
commit 3bf7e346fc
4 changed files with 15 additions and 1 deletions
@@ -664,6 +664,11 @@ if current_platform.is_mps():

    fuse_scale_shift_kernel = fuse_scale_shift_kernel_native

+if current_platform.is_musa():
+    from .torch_fallback import fuse_scale_shift_kernel_native
+
+    fuse_scale_shift_kernel = fuse_scale_shift_kernel_native
+
 if current_platform.is_cpu():
    from .torch_fallback import (
        fuse_scale_shift_kernel_native,