From b1b30afd4e47113d02cc2552582d501ad547ba95 Mon Sep 17 00:00:00 2001
From: taoxiangdong <taoxiangdong1@huawei.com>
Date: Wed, 16 Dec 2020 14:48:51 +0800
Subject: [PATCH] update cmakelist

---
 CMakeLists.txt    | 2 +-
 ge/CMakeLists.txt | 6 ++++++
 2 files changed, 7 insertions(+), 1 deletion(-)

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} $<TARGET_OBJECTS:msprofiler_fwk>)