| @@ -21,6 +21,13 @@ set(ASCEND_ATC_DIR ${ASCEND_DIR}/atc/lib64) | |||||
| set(ASCEND_ACL_DIR ${ASCEND_DIR}/acllib/lib64) | set(ASCEND_ACL_DIR ${ASCEND_DIR}/acllib/lib64) | ||||
| set(STATIC_ACL_LIB ${ASCEND_ACL_DIR}) | set(STATIC_ACL_LIB ${ASCEND_ACL_DIR}) | ||||
| set(ASCEND_MS_RUNTIME_PATH ${ASCEND_RUNTIME_DIR} ${ASCEND_ACL_DIR} ${ASCEND_ATC_DIR}) | |||||
| set(ASCEND_MS_DRIVER_PATH ${ASCEND_DRIVER_DIR} ${ASCEND_DRIVER_COMMON_DIR}) | |||||
| set(ATLAS_RUNTIME_DIR ${ASCEND_DIR}/ascend-toolkit/latest/fwkacllib/lib64) | |||||
| set(ATLAS_ACL_DIR ${ASCEND_DIR}/ascend-toolkit/latest/acllib/lib64) | |||||
| set(ATLAS_ATC_DIR ${ASCEND_DIR}/ascend-toolkit/latest/atc/lib64) | |||||
| set(ATLAS_MS_RUNTIME_PATH ${ATLAS_RUNTIME_DIR} ${ATLAS_ACL_DIR} ${ATLAS_ATC_DIR}) | |||||
| option(ENABLE_OPEN_SRC "Enable graphengine compile in opensource." FALSE) | option(ENABLE_OPEN_SRC "Enable graphengine compile in opensource." FALSE) | ||||
| if (ENABLE_OPEN_SRC) | if (ENABLE_OPEN_SRC) | ||||
| @@ -129,14 +136,6 @@ if (ENABLE_OPEN_SRC) | |||||
| #add_subdirectory(metadef/graph) | #add_subdirectory(metadef/graph) | ||||
| #add_subdirectory(metadef/register) | #add_subdirectory(metadef/register) | ||||
| elseif (ENABLE_D OR ENABLE_ACL) | elseif (ENABLE_D OR ENABLE_ACL) | ||||
| set(ASCEND_MS_RUNTIME_PATH ${ASCEND_RUNTIME_DIR} ${ASCEND_ACL_DIR} ${ASCEND_ATC_DIR}) | |||||
| set(ASCEND_MS_DRIVER_PATH ${ASCEND_DRIVER_DIR} ${ASCEND_DRIVER_COMMON_DIR}) | |||||
| set(ATLAS_RUNTIME_DIR ${ASCEND_DIR}/ascend-toolkit/latest/fwkacllib/lib64) | |||||
| set(ATLAS_ACL_DIR ${ASCEND_DIR}/ascend-toolkit/latest/acllib/lib64) | |||||
| set(ATLAS_ATC_DIR ${ASCEND_DIR}/ascend-toolkit/latest/atc/lib64) | |||||
| set(ATLAS_MS_RUNTIME_PATH ${ATLAS_RUNTIME_DIR} ${ATLAS_ACL_DIR} ${ATLAS_ATC_DIR}) | |||||
| # compiling with MindSpore | # compiling with MindSpore | ||||
| include(cmake/external_libs/protobuf_static.cmake) | include(cmake/external_libs/protobuf_static.cmake) | ||||
| include(cmake/external_libs/protoc.cmake) | include(cmake/external_libs/protoc.cmake) | ||||
| @@ -158,11 +157,18 @@ elseif (ENABLE_D OR ENABLE_ACL) | |||||
| set(METADEF_DIR ${CMAKE_CURRENT_LIST_DIR}/metadef) | set(METADEF_DIR ${CMAKE_CURRENT_LIST_DIR}/metadef) | ||||
| add_subdirectory(metadef) | add_subdirectory(metadef) | ||||
| elseif(ENABLE_MS_TESTCASE) | |||||
| elseif(ENABLE_MS_TESTCASES) | |||||
| include(cmake/external_libs/protobuf_static.cmake) | include(cmake/external_libs/protobuf_static.cmake) | ||||
| include(cmake/external_libs/protoc.cmake) | |||||
| include(cmake/external_libs/securec.cmake) | include(cmake/external_libs/securec.cmake) | ||||
| include(cmake/FindModule.cmake) | |||||
| include(cmake/intf_pub_linux.cmake) | include(cmake/intf_pub_linux.cmake) | ||||
| # common libraries | |||||
| find_module(slog libslog.so ${ASCEND_MS_DRIVER_PATH}) | |||||
| find_module(error_manager liberror_manager.so ${ASCEND_MS_RUNTIME_PATH} ${ATLAS_MS_RUNTIME_PATH}) | |||||
| find_module(static_mmpa libmmpa.a ${ASCEND_MS_RUNTIME_PATH} ${ATLAS_MS_RUNTIME_PATH}) | |||||
| set(METADEF_DIR ${CMAKE_CURRENT_LIST_DIR}/metadef) | set(METADEF_DIR ${CMAKE_CURRENT_LIST_DIR}/metadef) | ||||
| add_subdirectory(metadef) | add_subdirectory(metadef) | ||||
| else() | else() | ||||