rename infer_ext to dlinfer (#15)

CyCle1024 · web-flow · commit 8511b88c31d1 · 2024-08-22T15:19:48.000+08:00
diff --git a/lmdeploy/pytorch/kernels/ascend/apply_rotary_pos_emb.py b/lmdeploy/pytorch/kernels/ascend/apply_rotary_pos_emb.py
@@ -1,5 +1,5 @@
 # Copyright (c) OpenMMLab. All rights reserved.
-import infer_ext.ops as ext_ops
+import dlinfer.ops as ext_ops
 from torch import Tensor
 
 
diff --git a/lmdeploy/pytorch/kernels/ascend/fill_kv_cache.py b/lmdeploy/pytorch/kernels/ascend/fill_kv_cache.py
@@ -1,5 +1,5 @@
 # Copyright (c) OpenMMLab. All rights reserved.
-import infer_ext.ops as ext_ops
+import dlinfer.ops as ext_ops
 from torch import Tensor
 
 
diff --git a/lmdeploy/pytorch/kernels/ascend/fused_rotary_emb.py b/lmdeploy/pytorch/kernels/ascend/fused_rotary_emb.py
@@ -1,5 +1,5 @@
 # Copyright (c) OpenMMLab. All rights reserved.
-import infer_ext.ops as ext_ops
+import dlinfer.ops as ext_ops
 import torch
 from torch import Tensor
 
diff --git a/lmdeploy/pytorch/kernels/ascend/moe_gating_topk_softmax.py b/lmdeploy/pytorch/kernels/ascend/moe_gating_topk_softmax.py
@@ -1,5 +1,5 @@
 # Copyright (c) OpenMMLab. All rights reserved.
-import infer_ext.ops as ext_ops
+import dlinfer.ops as ext_ops
 import torch
 from torch import Tensor
 
diff --git a/lmdeploy/pytorch/kernels/ascend/paged_attention_fwd.py b/lmdeploy/pytorch/kernels/ascend/paged_attention_fwd.py
@@ -1,5 +1,5 @@
 # Copyright (c) OpenMMLab. All rights reserved.
-import infer_ext.ops as ext_ops
+import dlinfer.ops as ext_ops
 import torch
 from torch import Tensor
 
diff --git a/lmdeploy/pytorch/kernels/ascend/rms_norm.py b/lmdeploy/pytorch/kernels/ascend/rms_norm.py
@@ -1,5 +1,5 @@
 # Copyright (c) OpenMMLab. All rights reserved.
-import infer_ext.ops as ext_ops
+import dlinfer.ops as ext_ops
 from torch import Tensor
 
 
