diff --git a/lmdeploy/pytorch/backends/dlinfer/awq_modules.py b/lmdeploy/pytorch/backends/dlinfer/awq_modules.py
index 72badf7d7c..7eb767f62f 100644
--- a/lmdeploy/pytorch/backends/dlinfer/awq_modules.py
+++ b/lmdeploy/pytorch/backends/dlinfer/awq_modules.py
@@ -3,7 +3,7 @@
 
 import torch
 
-from lmdeploy.pytorch.kernels.ascend import awq_linear
+from lmdeploy.pytorch.kernels.dlinfer import awq_linear
 
 from ..awq_modules import LinearW4A16Builder, LinearW4A16Impl