diff --git a/CMakeLists.txt b/CMakeLists.txt
index d45b223693cc09c6c76f44a6a6d5faf178b4e5cc..0fe0fdc24106f324a044fe41473651c2ae2777aa 100755
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -18,20 +18,26 @@ if (ENABLE_OPEN_SRC)
         message("Download packages from PKG server")
     endif()
 
+    if (NOT DEFINED ASCEND_OPENSDK_DIR)
+        set(ASCEND_OPENSDK_DIR $ENV{ASCEND_CUSTOM_PATH}/opensdk/opensdk)
+    endif()
+
     include(${CMAKE_CURRENT_LIST_DIR}/cmake/nlohmann_json.cmake)
     include(${CMAKE_CURRENT_LIST_DIR}/cmake/secure_c.cmake)
     include(${CMAKE_CURRENT_LIST_DIR}/cmake/tensorflow.cmake)
-    include_directories(${CMAKE_CURRENT_LIST_DIR})
-    include_directories(${CMAKE_CURRENT_LIST_DIR}/inc)
-    include_directories(${CMAKE_CURRENT_LIST_DIR}/inc/toolchain)
+    include_directories(${CMAKE_CURRENT_LIST_DIR}) 
     include_directories(${CMAKE_CURRENT_LIST_DIR}/inc/toolchain/tuning_tool)
-    include_directories(${CMAKE_CURRENT_LIST_DIR}/inc/external)
-    include_directories(${CMAKE_CURRENT_LIST_DIR}/inc/parser/inc)
-    include_directories(${CMAKE_CURRENT_LIST_DIR}/inc/parser/inc/external)
-    include_directories(${CMAKE_CURRENT_LIST_DIR}/inc/graphengine/inc)
-    include_directories(${CMAKE_CURRENT_LIST_DIR}/inc/graphengine/inc/external)
-    include_directories(${CMAKE_CURRENT_LIST_DIR}/inc/metadef/inc)
-    include_directories(${CMAKE_CURRENT_LIST_DIR}/inc/metadef/inc/external)
+    include_directories(${ASCEND_OPENSDK_DIR}/include)
+    include_directories(${ASCEND_OPENSDK_DIR}/include/slog)
+    include_directories(${ASCEND_OPENSDK_DIR}/include/runtime)
+    include_directories(${ASCEND_OPENSDK_DIR}/include/parser)
+    include_directories(${ASCEND_OPENSDK_DIR}/include/msprof)
+    include_directories(${ASCEND_OPENSDK_DIR}/include/ascendcl/external/)
+    include_directories(${ASCEND_OPENSDK_DIR}/include/parser/external)
+    include_directories(${ASCEND_OPENSDK_DIR}/include/air)
+    include_directories(${ASCEND_OPENSDK_DIR}/include/air/external)
+    include_directories(${ASCEND_OPENSDK_DIR}/include/metadef)
+    include_directories(${ASCEND_OPENSDK_DIR}/include/metadef/external)
 
     if (NOT EXISTS ${CMAKE_CURRENT_LIST_DIR}/tools/COMPILE_FLAGS OR NOT EXISTS
             ${CMAKE_CURRENT_LIST_DIR}/tools/LINK_FLAGS OR NOT EXISTS