File tree Expand file tree Collapse file tree 1 file changed +4
-1
lines changed
deeplink_ext/interntrain_ops Expand file tree Collapse file tree 1 file changed +4
-1
lines changed Original file line number Diff line number Diff line change 5
5
platform_type = deeplink_ext_get_platform_type ()
6
6
if platform_type == PlatformType .TORCH_NPU :
7
7
# from ._mixed_rms_norm_npu import MixedFusedRMSNorm
8
+ # Due to the accuracy problem of the npu fused operator, a torch combination is used as an alternative.
8
9
from .rms_norm_fallback import MixedRMSNormTorch as MixedFusedRMSNorm
9
10
elif platform_type == PlatformType .TORCH_DIPU :
10
- from ._mixed_rms_norm_dipu import MixedFusedRMSNorm
11
+ # from ._mixed_rms_norm_dipu import MixedFusedRMSNorm
12
+ # Due to the accuracy problem of the npu fused operator, a torch combination is used as an alternative.
13
+ from .rms_norm_fallback import MixedRMSNormTorch as MixedFusedRMSNorm
11
14
else :
12
15
raise ImportError
13
16
You can’t perform that action at this time.
0 commit comments