fix some bugs

DeepLink-org · Nov 6, 2024 · 10fa4ba · 10fa4ba
1 parent a0ad15b
commit 10fa4ba
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/deeplink_ext/ops/rms_norm/internevo_rms_norm.py b/deeplink_ext/ops/rms_norm/internevo_rms_norm.py
@@ -6,7 +6,7 @@
 if platform_type == PlatformType.TORCH_DIPU:
     # from ._mixed_rms_norm_dipu import MixedFusedRMSNorm
     # Due to the accuracy problem of the npu fused operator, a torch combination is used as an alternative.
-    from .interntrain_rms_norm_fallback import MixedRMSNormTorch as MixedFusedRMSNorm
+    from .internevo_rms_norm_fallback import MixedRMSNormTorch as MixedFusedRMSNorm
 else:
     raise ImportError
 

diff --git a/deeplink_ext/ops/rotary_embedding/interntrain_rotary_embedding.py b/deeplink_ext/ops/rotary_embedding/interntrain_rotary_embedding.py
@@ -4,7 +4,7 @@
 
 platform_type = deeplink_ext_get_platform_type()
 if platform_type == PlatformType.TORCH_DIPU:
-    from ._rotary_embedding_dipu import ApplyRotaryEmb, ApplyRotaryEmbQKV_
+    from .interntrain_rotary_embedding_dipu import ApplyRotaryEmb, ApplyRotaryEmbQKV_
 else:
     raise ImportError