|
|
@@ -20,27 +20,57 @@ if (ENABLE_OPEN_SRC) |
|
|
|
include(cmake/FindModule.cmake) |
|
|
|
include(cmake/intf_pub_linux.cmake) |
|
|
|
|
|
|
|
if(DEFINED ENV{ASCEND_CUSTOM_PATH}) |
|
|
|
set(ASCEND_DIR $ENV{ASCEND_CUSTOM_PATH}) |
|
|
|
# for CPU/GPU mode, find c_sec and slog from local prebuild |
|
|
|
if(NOT ENABLE_D AND NOT GE_ONLY) |
|
|
|
set(GE_PREBUILD_PATH ${GE_SOURCE_DIR}/third_party/prebuild/${CMAKE_HOST_SYSTEM_PROCESSOR}) |
|
|
|
find_module(slog libslog.so ${GE_PREBUILD_PATH}) |
|
|
|
# if D_LINK_PATH is set in environment variables, search libraries in given path |
|
|
|
elseif(DEFINED ENV{D_LINK_PATH}) |
|
|
|
# D_LINK_PATH is set |
|
|
|
set(GE_LIB_PATH $ENV{D_LINK_PATH}) |
|
|
|
set(GE_SYS_ARCH "") |
|
|
|
if(CMAKE_HOST_SYSTEM_PROCESSOR MATCHES "x86_64") |
|
|
|
# x86 ubuntu |
|
|
|
set(GE_SYS_ARCH "x86_64") |
|
|
|
elseif(CMAKE_HOST_SYSTEM_PROCESSOR MATCHES "aarch64") |
|
|
|
# arm euleros |
|
|
|
set(GE_SYS_ARCH "aarch64") |
|
|
|
else() |
|
|
|
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 libslog.so ${GE_LIB_PATH}) |
|
|
|
find_module(mmpa libmmpa.so ${GE_LIB_PATH}) |
|
|
|
find_module(msprof libmsprof.so ${GE_LIB_PATH}) |
|
|
|
find_module(hccl libhccl.so ${GE_LIB_PATH}) |
|
|
|
find_module(adump_server libadump_server.a ${GE_LIB_PATH}) |
|
|
|
find_module(runtime libruntime.so ${GE_LIB_PATH}) |
|
|
|
find_module(runtime_compile libruntime_compile.so ${GE_LIB_PATH}) |
|
|
|
find_module(resource libresource.so ${GE_LIB_PATH}) |
|
|
|
find_module(error_manager liberror_manager.so ${GE_LIB_PATH}) |
|
|
|
find_module(ascend_hal_stub libascend_hal.so ${GE_LIB_PATH}) |
|
|
|
else() |
|
|
|
set(ASCEND_DIR /usr/local/Ascend) |
|
|
|
if(DEFINED ENV{ASCEND_CUSTOM_PATH}) |
|
|
|
set(ASCEND_DIR $ENV{ASCEND_CUSTOM_PATH}) |
|
|
|
else() |
|
|
|
set(ASCEND_DIR /usr/local/Ascend) |
|
|
|
endif() |
|
|
|
set(ASCEND_DRIVER_DIR ${ASCEND_DIR}/driver/lib64) |
|
|
|
set(ASCEND_DRIVER_COMMON_DIR ${ASCEND_DIR}/driver/lib64/common) |
|
|
|
set(ASCEND_RUNTIME_DIR ${ASCEND_DIR}/fwkacllib/lib64) |
|
|
|
set(ASCEND_ATC_DIR ${ASCEND_DIR}/atc/lib64) |
|
|
|
find_module(slog libslog.so ${ASCEND_DRIVER_DIR}) |
|
|
|
find_module(mmpa libmmpa.so ${ASCEND_DRIVER_DIR}) |
|
|
|
find_module(msprof libmsprof.so ${ASCEND_DRIVER_DIR}) |
|
|
|
find_module(hccl libhccl.so ${ASCEND_RUNTIME_DIR}) |
|
|
|
find_module(adump_server libadump_server.a ${ASCEND_RUNTIME_DIR}) |
|
|
|
find_module(runtime libruntime.so ${ASCEND_RUNTIME_DIR}) |
|
|
|
find_module(runtime_compile libruntime_compile.so ${ASCEND_ATC_DIR}) |
|
|
|
find_module(resource libresource.so ${ASCEND_RUNTIME_DIR}) |
|
|
|
find_module(error_manager liberror_manager.so ${ASCEND_RUNTIME_DIR}) |
|
|
|
find_module(ascend_hal_stub libascend_hal.so ${ASCEND_DRIVER_DIR}) |
|
|
|
endif() |
|
|
|
|
|
|
|
set(ASCEND_DRIVER_DIR ${ASCEND_DIR}/driver/lib64) |
|
|
|
set(ASCEND_DRIVER_COMMON_DIR ${ASCEND_DIR}/driver/lib64/common) |
|
|
|
set(ASCEND_RUNTIME_DIR ${ASCEND_DIR}/fwkacllib/lib64) |
|
|
|
set(ASCEND_ATC_DIR ${ASCEND_DIR}/atc/lib64) |
|
|
|
find_module(slog libslog.so ${ASCEND_DRIVER_DIR}) |
|
|
|
find_module(mmpa libmmpa.so ${ASCEND_DRIVER_DIR}) |
|
|
|
find_module(msprof libmsprof.so ${ASCEND_DRIVER_DIR}) |
|
|
|
find_module(hccl libhccl.so ${ASCEND_RUNTIME_DIR}) |
|
|
|
find_module(adump_server libadump_server.a ${ASCEND_RUNTIME_DIR}) |
|
|
|
find_module(runtime libruntime.so ${ASCEND_RUNTIME_DIR}) |
|
|
|
find_module(runtime_compile libruntime_compile.so ${ASCEND_ATC_DIR}) |
|
|
|
find_module(resource libresource.so ${ASCEND_RUNTIME_DIR}) |
|
|
|
find_module(error_manager liberror_manager.so ${ASCEND_RUNTIME_DIR}) |
|
|
|
find_module(ascend_hal_stub libascend_hal.so ${ASCEND_DRIVER_DIR}) |
|
|
|
|
|
|
|
set(METADEF_DIR ${CMAKE_CURRENT_LIST_DIR}/metadef) |
|
|
|
set(PARSER_DIR ${CMAKE_CURRENT_LIST_DIR}/parser) |
|
|
|
set(GE_DEPEND_DIR ${CMAKE_CURRENT_LIST_DIR}/..) |
|
|
|