diff --git a/CMakeLists.txt b/CMakeLists.txt
index 2c0aa4b..a413ea3 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -50,10 +50,15 @@ if (ENABLE_OPEN_SRC)
             set(ASCEND_DIR /usr/local/Ascend)
         endif()
 
-        set(ASCEND_ATC_DIR ${ASCEND_DIR}/atc/lib64)
-
-        find_module(slog libalog.so ${ASCEND_ATC_DIR})
-        find_module(static_mmpa libmmpa.a ${ASCEND_ATC_DIR})
+        if(DEFINED ENV{ALL_IN_ONE_ENABLE})
+            set(ASCEND_COMPILER_DIR ${ASCEND_DIR}/compiler/lib64)
+            find_module(slog libalog.so ${ASCEND_COMPILER_DIR})
+            find_module(static_mmpa libmmpa.a ${ASCEND_COMPILER_DIR})
+        else()
+            set(ASCEND_ATC_DIR ${ASCEND_DIR}/atc/lib64)
+            find_module(slog libalog.so ${ASCEND_ATC_DIR})
+            find_module(static_mmpa libmmpa.a ${ASCEND_ATC_DIR})
+        endif()
     endif()
 
     if (NOT DEFINED METADEF_DIR)
diff --git a/build.sh b/build.sh
index a4a2a87..586bf99 100644
--- a/build.sh
+++ b/build.sh
@@ -255,14 +255,14 @@ generate_package_for_cann()
   cd "${BASEPATH}"
 
   PARSER_LIB_PATH="lib"
-  CMPL_PATH="compiler/lib64"
+  COMPILER_PATH="compiler/lib64"
 
   COMMON_LIB=("libgraph.so" "libregister.so" "liberror_manager.so")
   PARSER_LIB=("lib_caffe_parser.so" "libfmk_onnx_parser.so" "libfmk_parser.so" "libparser_common.so")
 
-  rm -rf ${OUTPUT_PATH:?}/${CMPL_PATH}/
+  rm -rf ${OUTPUT_PATH:?}/${COMPILER_PATH}/
 
-  mk_dir "${OUTPUT_PATH}/${CMPL_PATH}"
+  mk_dir "${OUTPUT_PATH}/${COMPILER_PATH}"
 
   find output/ -name parser_lib.tar -exec rm {} \;
 
@@ -270,15 +270,15 @@ generate_package_for_cann()
 
   for lib in "${PARSER_LIB[@]}";
   do
-    find ${OUTPUT_PATH}/${PARSER_LIB_PATH} -maxdepth 1 -name "$lib" -exec cp -f {} ${OUTPUT_PATH}/${CMPL_PATH} \;
+    find ${OUTPUT_PATH}/${PARSER_LIB_PATH} -maxdepth 1 -name "$lib" -exec cp -f {} ${OUTPUT_PATH}/${COMPILER_PATH} \;
   done
 
   for lib in "${COMMON_LIB[@]}";
   do
-    find ${OUTPUT_PATH}/${PARSER_LIB_PATH} -maxdepth 1 -name "$lib" -exec cp -f {} ${OUTPUT_PATH}/${CMPL_PATH} \;
+    find ${OUTPUT_PATH}/${PARSER_LIB_PATH} -maxdepth 1 -name "$lib" -exec cp -f {} ${OUTPUT_PATH}/${COMPILER_PATH} \;
   done
 
-  find ${OUTPUT_PATH}/${PARSER_LIB_PATH} -maxdepth 1 -name "libc_sec.so" -exec cp -f {} ${OUTPUT_PATH}/${CMPL_PATH} \;
+  find ${OUTPUT_PATH}/${PARSER_LIB_PATH} -maxdepth 1 -name "libc_sec.so" -exec cp -f {} ${OUTPUT_PATH}/${COMPILER_PATH} \;
 
   tar -cf parser_lib.tar compiler
 }