| @@ -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 | ||||