diff --git a/CMakeLists.txt b/CMakeLists.txt
index 0ca9fcee..8aa8a360 100755
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -76,21 +76,21 @@ elseif(DEFINED ENV{D_LINK_PATH})
 else()
     # Ascend mode
     if(DEFINED ENV{ASCEND_CUSTOM_PATH})
-        set(HIAI_INSTALLED_DIR $ENV{D_LINK_PATH})
+        set(ASCEND_DIR $ENV{ASCEND_CUSTOM_PATH})
     else()
-        set(HIAI_INSTALLED_DIR /usr/local/Ascend)
+        set(ASCEND_DIR /usr/local/Ascend)
     endif()
-    set(HIAI_DRIVER_DIR ${HIAI_INSTALLED_DIR}/driver/lib64/common)
-    set(HIAI_RUNTIME_DIR ${HIAI_INSTALLED_DIR}/fwkacllib/lib64)
-    find_library(c_sec libc_sec.so ${HIAI_DRIVER_DIR})
-    find_library(slog libslog.so ${HIAI_DRIVER_DIR})
-    find_library(mmpa libmmpa.so ${HIAI_DRIVER_DIR})
-    find_library(msprof libmsprof.so ${HIAI_DRIVER_DIR})
+    set(ASCEND_DRIVER_DIR ${ASCEND_DIR}/driver/lib64/common)
+    set(ASCEND_RUNTIME_DIR ${ASCEND_DIR}/fwkacllib/lib64)
+    find_library(c_sec libc_sec.so ${ASCEND_DRIVER_DIR})
+    find_library(slog libslog.so ${ASCEND_DRIVER_DIR})
+    find_library(mmpa libmmpa.so ${ASCEND_DRIVER_DIR})
+    find_library(msprof libmsprof.so ${ASCEND_DRIVER_DIR})
 
-    find_library(hccl libhccl.so ${HIAI_RUNTIME_DIR})
-    find_library(runtime libruntime.so ${HIAI_RUNTIME_DIR})
-    find_library(register libregister.so ${HIAI_RUNTIME_DIR})
-    find_library(resource libresource.so ${HIAI_RUNTIME_DIR})
+    find_library(hccl libhccl.so ${ASCEND_RUNTIME_DIR})
+    find_library(runtime libruntime.so ${ASCEND_RUNTIME_DIR})
+    find_library(register libregister.so ${ASCEND_RUNTIME_DIR})
+    find_library(resource libresource.so ${ASCEND_RUNTIME_DIR})
 endif()
 
 # add compile flags