|
|
|
@ -66,30 +66,30 @@ if (ENABLE_OPEN_SRC)
|
|
|
|
|
set(ASCEND_ATC_DIR ${ASCEND_DIR}/atc/lib64)
|
|
|
|
|
set(ASCEND_ACL_DIR ${ASCEND_DIR}/acllib/lib64)
|
|
|
|
|
if(PLATFORM STREQUAL "train")
|
|
|
|
|
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(slog libslog.so ${ASCEND_ATC_DIR})
|
|
|
|
|
find_module(mmpa libmmpa.so ${ASCEND_ATC_DIR})
|
|
|
|
|
find_module(msprof libmsprof.so ${ASCEND_DRIVER_COMMON_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(resource libresource.so ${ASCEND_RUNTIME_DIR})
|
|
|
|
|
find_module(error_manager liberror_manager.so ${ASCEND_RUNTIME_DIR})
|
|
|
|
|
find_module(msprofiler libmsprofiler.a ${ASCEND_RUNTIME_DIR})
|
|
|
|
|
find_module(ascend_hal_stub libascend_hal.so ${ASCEND_DRIVER_DIR})
|
|
|
|
|
find_module(ascend_hal_stub libascend_hal.so ${ASCEND_DRIVER_DIR}/driver)
|
|
|
|
|
if(PRODUCT STREQUAL "flr3")
|
|
|
|
|
message(FATAL_ERROR "This platform is not supported in train mode, build terminated")
|
|
|
|
|
endif()
|
|
|
|
|
elseif(PLATFORM STREQUAL "inference")
|
|
|
|
|
find_module(slog libslog.so ${ASCEND_ATC_DIR})
|
|
|
|
|
find_module(mmpa libmmpa.so ${ASCEND_ATC_DIR})
|
|
|
|
|
find_module(adump_server libadump_server.a ${ASCEND_ACL_DIR})
|
|
|
|
|
find_module(runtime libruntime.so ${ASCEND_ACL_DIR})
|
|
|
|
|
find_module(runtime_compile libruntime_compile.so ${ASCEND_ATC_DIR})
|
|
|
|
|
find_module(resource libresource.so ${ASCEND_ACL_DIR})
|
|
|
|
|
find_module(error_manager liberror_manager.so ${ASCEND_ACL_DIR})
|
|
|
|
|
find_module(resource libresource.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(msprofiler libmsprofiler.a ${ASCEND_ACL_DIR})
|
|
|
|
|
find_module(ascendcl_static libascendcl.a ${ASCEND_ACL_DIR})
|
|
|
|
|
find_module(slog libslog.so ${ASCEND_ATC_DIR})
|
|
|
|
|
find_module(mmpa libmmpa.so ${ASCEND_ATC_DIR})
|
|
|
|
|
if(NOT PRODUCT STREQUAL "flr3")
|
|
|
|
|
find_module(ascend_hal_stub libascend_hal.so ${ASCEND_DRIVER_DIR})
|
|
|
|
|
find_module(msprof libmsprof.so ${ASCEND_DRIVER_DIR})
|
|
|
|
|