|
|
@@ -21,6 +21,13 @@ set(ASCEND_ATC_DIR ${ASCEND_DIR}/atc/lib64) |
|
|
|
set(ASCEND_ACL_DIR ${ASCEND_DIR}/acllib/lib64) |
|
|
|
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) |
|
|
|
|
|
|
|
if (ENABLE_OPEN_SRC) |
|
|
@@ -129,14 +136,6 @@ if (ENABLE_OPEN_SRC) |
|
|
|
#add_subdirectory(metadef/graph) |
|
|
|
#add_subdirectory(metadef/register) |
|
|
|
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 |
|
|
|
include(cmake/external_libs/protobuf_static.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) |
|
|
|
add_subdirectory(metadef) |
|
|
|
elseif(ENABLE_MS_TESTCASE) |
|
|
|
elseif(ENABLE_MS_TESTCASES) |
|
|
|
include(cmake/external_libs/protobuf_static.cmake) |
|
|
|
include(cmake/external_libs/protoc.cmake) |
|
|
|
include(cmake/external_libs/securec.cmake) |
|
|
|
include(cmake/FindModule.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) |
|
|
|
add_subdirectory(metadef) |
|
|
|
else() |
|
|
|