diff --git a/CMakeLists.txt b/CMakeLists.txt index 0bd4b913..bea12fcc 100755 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -92,7 +92,7 @@ if (ENABLE_OPEN_SRC) 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_fwk libmsprofiler_fwk.a ${ASCEND_RUNTIME_DIR}) + #find_module(msprofiler_fwk libmsprofiler_fwk.a ${ASCEND_RUNTIME_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") diff --git a/ge/CMakeLists.txt b/ge/CMakeLists.txt index 015e7a1d..6fe43a1c 100755 --- a/ge/CMakeLists.txt +++ b/ge/CMakeLists.txt @@ -620,6 +620,12 @@ set(INFER_SRC_LIST # add_library(msprofiler_fwk OBJECT ${msprof_file}) #endif() +if (ENABLE_OPEN_SRC) + add_library(msprofiler_fwk STATIC IMPORTED) + set_target_properties(msprofiler_fwk PROPERTIES + IMPORTED_LOCATION ${ASCEND_RUNTIME_DIR}/libmsprofiler_fwk.a) +endif() + if (NOT ENABLE_D AND NOT ENABLE_ACL AND NOT ENABLE_MS_TESTCASES) ############ libge_runner.so ############ add_library(ge_runner SHARED ${TRAIN_SRC_LIST} ${PROTO_SRCS} ${PROTO_CLIENT_SRCS} $)