|
@@ -41,6 +41,7 @@ if (ENABLE_OPEN_SRC) |
|
|
message(FATAL_ERROR "Running on a unsupported architecture: ${SYSTEM_TYPE}, build terminated") |
|
|
message(FATAL_ERROR "Running on a unsupported architecture: ${SYSTEM_TYPE}, build terminated") |
|
|
endif() |
|
|
endif() |
|
|
set(GE_LIB_PATH ${GE_LIB_PATH}/${GE_SYS_ARCH}) |
|
|
set(GE_LIB_PATH ${GE_LIB_PATH}/${GE_SYS_ARCH}) |
|
|
|
|
|
set(STATIC_ACL_LIB ${GE_LIB_PATH}) |
|
|
find_module(slog libslog.so ${GE_LIB_PATH}) |
|
|
find_module(slog libslog.so ${GE_LIB_PATH}) |
|
|
find_module(mmpa libmmpa.so ${GE_LIB_PATH}) |
|
|
find_module(mmpa libmmpa.so ${GE_LIB_PATH}) |
|
|
find_module(msprof libmsprof.so ${GE_LIB_PATH}) |
|
|
find_module(msprof libmsprof.so ${GE_LIB_PATH}) |
|
@@ -53,7 +54,7 @@ if (ENABLE_OPEN_SRC) |
|
|
find_module(ascend_hal_stub libascend_hal.so ${GE_LIB_PATH}) |
|
|
find_module(ascend_hal_stub libascend_hal.so ${GE_LIB_PATH}) |
|
|
find_module(error_manager_static liberror_manager.a ${GE_LIB_PATH}) |
|
|
find_module(error_manager_static liberror_manager.a ${GE_LIB_PATH}) |
|
|
find_module(msprofiler libmsprofiler.a ${GE_LIB_PATH}) |
|
|
find_module(msprofiler libmsprofiler.a ${GE_LIB_PATH}) |
|
|
find_module(ascendcl_static libascendcl.a ${GE_LIB_PATH}) |
|
|
|
|
|
|
|
|
#find_module(ascendcl_static libascendcl.a ${GE_LIB_PATH}) |
|
|
else() |
|
|
else() |
|
|
if(DEFINED ENV{ASCEND_CUSTOM_PATH}) |
|
|
if(DEFINED ENV{ASCEND_CUSTOM_PATH}) |
|
|
set(ASCEND_DIR $ENV{ASCEND_CUSTOM_PATH}) |
|
|
set(ASCEND_DIR $ENV{ASCEND_CUSTOM_PATH}) |
|
@@ -66,6 +67,7 @@ if (ENABLE_OPEN_SRC) |
|
|
set(ASCEND_RUNTIME_DIR ${ASCEND_DIR}/fwkacllib/lib64) |
|
|
set(ASCEND_RUNTIME_DIR ${ASCEND_DIR}/fwkacllib/lib64) |
|
|
set(ASCEND_ATC_DIR ${ASCEND_DIR}/atc/lib64) |
|
|
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}) |
|
|
find_module(slog libslog.so ${ASCEND_ATC_DIR}) |
|
|
find_module(slog libslog.so ${ASCEND_ATC_DIR}) |
|
|
find_module(mmpa libmmpa.so ${ASCEND_ATC_DIR}) |
|
|
find_module(mmpa libmmpa.so ${ASCEND_ATC_DIR}) |
|
|
if(PLATFORM STREQUAL "train") |
|
|
if(PLATFORM STREQUAL "train") |
|
@@ -88,7 +90,7 @@ if (ENABLE_OPEN_SRC) |
|
|
find_module(error_manager liberror_manager.so ${ASCEND_ATC_DIR}) |
|
|
find_module(error_manager liberror_manager.so ${ASCEND_ATC_DIR}) |
|
|
find_module(error_manager_static liberror_manager.a ${ASCEND_ACL_DIR}) |
|
|
find_module(error_manager_static liberror_manager.a ${ASCEND_ACL_DIR}) |
|
|
find_module(msprofiler libmsprofiler.a ${ASCEND_ACL_DIR}) |
|
|
find_module(msprofiler libmsprofiler.a ${ASCEND_ACL_DIR}) |
|
|
find_module(ascendcl_static libascendcl.a ${ASCEND_ACL_DIR}) |
|
|
|
|
|
|
|
|
#find_module(ascendcl_static libascendcl.a ${ASCEND_ACL_DIR}) |
|
|
if(PRODUCT STREQUAL "flr3") |
|
|
if(PRODUCT STREQUAL "flr3") |
|
|
find_module(msprof libmsprof.so ${ASCEND_DRIVER_SHARE_DIR}) |
|
|
find_module(msprof libmsprof.so ${ASCEND_DRIVER_SHARE_DIR}) |
|
|
elseif(PRODUCT STREQUAL "flr1") |
|
|
elseif(PRODUCT STREQUAL "flr1") |
|
@@ -111,7 +113,7 @@ if (ENABLE_OPEN_SRC) |
|
|
find_module(error_manager_static liberror_manager.a ${ASCEND_ACL_DIR}) |
|
|
find_module(error_manager_static liberror_manager.a ${ASCEND_ACL_DIR}) |
|
|
find_module(msprofiler libmsprofiler.a ${ASCEND_ACL_DIR}) |
|
|
find_module(msprofiler libmsprofiler.a ${ASCEND_ACL_DIR}) |
|
|
find_module(ascend_hal_stub libascend_hal.so ${ASCEND_DRIVER_DIR}/driver) |
|
|
find_module(ascend_hal_stub libascend_hal.so ${ASCEND_DRIVER_DIR}/driver) |
|
|
find_module(ascendcl_static libascendcl.a ${ASCEND_ACL_DIR}) |
|
|
|
|
|
|
|
|
#find_module(ascendcl_static libascendcl.a ${ASCEND_ACL_DIR}) |
|
|
else() |
|
|
else() |
|
|
message(FATAL_ERROR "PLATFORM param is invalid, should be train or inference, build terminated") |
|
|
message(FATAL_ERROR "PLATFORM param is invalid, should be train or inference, build terminated") |
|
|
endif() |
|
|
endif() |
|
|