lxd-cumt
diff --git a/‎transformer_engine/__init__.py‎
Lines changed: 1 addition & 1 deletion b/‎transformer_engine/__init__.py‎
Lines changed: 1 addition & 1 deletion
diff --git a/‎…ore/backends/vendor/musa/musa_patches.py‎ ‎…gin/core/backends/vendor/musa/patches.py‎transformer_engine/plugin/core/backends/vendor/musa/musa_patches.py renamed to transformer_engine/plugin/core/backends/vendor/musa/patches.py b/‎…ore/backends/vendor/musa/musa_patches.py‎ ‎…gin/core/backends/vendor/musa/patches.py‎transformer_engine/plugin/core/backends/vendor/musa/musa_patches.py renamed to transformer_engine/plugin/core/backends/vendor/musa/patches.py
@@ -18,7 +18,7 @@
 
 # Apply MUSA (VENDOR) Patches, such as torch.cuda.device -> torch.musa.device
 try:
-    from .plugin.core.backends.vendor.musa.musa_patches import apply_patch as _musa_apply_patch
+    from .plugin.core.backends.vendor.musa.patches import apply_patch as _musa_apply_patch
 
     _musa_apply_patch()
     print("[TE-FL] MUSA patches applied")