diff --git a/CMakeLists.txt b/CMakeLists.txt index 6f46a1e..c9fe466 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -43,7 +43,11 @@ if (ENABLE_OPEN_SRC) message(FATAL_ERROR "Running on a unsupported architecture: ${SYSTEM_TYPE}, build terminated") endif() set(GE_LIB_PATH ${GE_LIB_PATH}/${GE_SYS_ARCH}) - find_module(slog libalog.so ${GE_LIB_PATH}) + if(ENABLE_OPEN_SRC) + find_module(slog libalog.so ${GE_LIB_PATH}) + else() + find_module(slog libascendalog.so ${GE_LIB_PATH}) + endif() find_module(static_mmpa libmmpa.a ${GE_LIB_PATH}) elseif(ENABLE_GE_COV OR ENABLE_GE_UT) message(STATUS "Runing on llt mode, no need to depend other component") @@ -58,7 +62,11 @@ if (ENABLE_OPEN_SRC) endif() set(ASCEND_COMPILER_DIR ${ASCEND_DIR}/compiler/lib64) - find_module(slog libalog.so ${ASCEND_COMPILER_DIR}) + if(ENABLE_OPEN_SRC) + find_module(slog libalog.so ${ASCEND_COMPILER_DIR}) + else() + find_module(slog libascendalog.so ${ASCEND_COMPILER_DIR}) + endif() find_module(static_mmpa libmmpa.a ${ASCEND_COMPILER_DIR}) endif()