|
@@ -76,21 +76,21 @@ elseif(DEFINED ENV{D_LINK_PATH}) |
|
|
else() |
|
|
else() |
|
|
# Ascend mode |
|
|
# Ascend mode |
|
|
if(DEFINED ENV{ASCEND_CUSTOM_PATH}) |
|
|
if(DEFINED ENV{ASCEND_CUSTOM_PATH}) |
|
|
set(HIAI_INSTALLED_DIR $ENV{D_LINK_PATH}) |
|
|
|
|
|
|
|
|
set(ASCEND_DIR $ENV{ASCEND_CUSTOM_PATH}) |
|
|
else() |
|
|
else() |
|
|
set(HIAI_INSTALLED_DIR /usr/local/Ascend) |
|
|
|
|
|
|
|
|
set(ASCEND_DIR /usr/local/Ascend) |
|
|
endif() |
|
|
endif() |
|
|
set(HIAI_DRIVER_DIR ${HIAI_INSTALLED_DIR}/driver/lib64/common) |
|
|
|
|
|
set(HIAI_RUNTIME_DIR ${HIAI_INSTALLED_DIR}/fwkacllib/lib64) |
|
|
|
|
|
find_library(c_sec libc_sec.so ${HIAI_DRIVER_DIR}) |
|
|
|
|
|
find_library(slog libslog.so ${HIAI_DRIVER_DIR}) |
|
|
|
|
|
find_library(mmpa libmmpa.so ${HIAI_DRIVER_DIR}) |
|
|
|
|
|
find_library(msprof libmsprof.so ${HIAI_DRIVER_DIR}) |
|
|
|
|
|
|
|
|
set(ASCEND_DRIVER_DIR ${ASCEND_DIR}/driver/lib64/common) |
|
|
|
|
|
set(ASCEND_RUNTIME_DIR ${ASCEND_DIR}/fwkacllib/lib64) |
|
|
|
|
|
find_library(c_sec libc_sec.so ${ASCEND_DRIVER_DIR}) |
|
|
|
|
|
find_library(slog libslog.so ${ASCEND_DRIVER_DIR}) |
|
|
|
|
|
find_library(mmpa libmmpa.so ${ASCEND_DRIVER_DIR}) |
|
|
|
|
|
find_library(msprof libmsprof.so ${ASCEND_DRIVER_DIR}) |
|
|
|
|
|
|
|
|
find_library(hccl libhccl.so ${HIAI_RUNTIME_DIR}) |
|
|
|
|
|
find_library(runtime libruntime.so ${HIAI_RUNTIME_DIR}) |
|
|
|
|
|
find_library(register libregister.so ${HIAI_RUNTIME_DIR}) |
|
|
|
|
|
find_library(resource libresource.so ${HIAI_RUNTIME_DIR}) |
|
|
|
|
|
|
|
|
find_library(hccl libhccl.so ${ASCEND_RUNTIME_DIR}) |
|
|
|
|
|
find_library(runtime libruntime.so ${ASCEND_RUNTIME_DIR}) |
|
|
|
|
|
find_library(register libregister.so ${ASCEND_RUNTIME_DIR}) |
|
|
|
|
|
find_library(resource libresource.so ${ASCEND_RUNTIME_DIR}) |
|
|
endif() |
|
|
endif() |
|
|
|
|
|
|
|
|
# add compile flags |
|
|
# add compile flags |
|
|