update cmakelist

pull/123/head
taoxiangdong 4 years ago
parent cb3cc7993e
commit 797a806e27

@ -96,8 +96,10 @@ if (ENABLE_OPEN_SRC)
find_module(msprof libmsprof.so ${ASCEND_DRIVER_SHARE_DIR}) find_module(msprof libmsprof.so ${ASCEND_DRIVER_SHARE_DIR})
endif() endif()
elseif(PLATFORM STREQUAL "all") elseif(PLATFORM STREQUAL "all")
find_module(msprof libmsprof.so ${ASCEND_DRIVER_DIR})#mdc dcdriver #mdc dcdriver
find_module(hccl libhccl.so ${ASCEND_RUNTIME_DIR})#mdc dcfwkack find_module(msprof libmsprof.so ${ASCEND_DRIVER_DIR})
#mdc dcfwkack
find_module(hccl libhccl.so ${ASCEND_RUNTIME_DIR})
find_module(adump_server libadump_server.a ${ASCEND_ACL_DIR}) find_module(adump_server libadump_server.a ${ASCEND_ACL_DIR})
find_module(runtime libruntime.so ${ASCEND_ACL_DIR}) find_module(runtime libruntime.so ${ASCEND_ACL_DIR})
find_module(runtime_compile libruntime_compile.so ${ASCEND_ATC_DIR}) find_module(runtime_compile libruntime_compile.so ${ASCEND_ATC_DIR})
@ -105,7 +107,8 @@ 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(ascend_hal_stub libascend_hal.so ${ASCEND_DRIVER_DIR}/driver)#mdc dcdriver/lib64/driver #mdc dcdriver/lib64/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")

Loading…
Cancel
Save