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

---
 ge/CMakeLists.txt | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/ge/CMakeLists.txt b/ge/CMakeLists.txt
index 59b804d8..306d4c98 100755
--- a/ge/CMakeLists.txt
+++ b/ge/CMakeLists.txt
@@ -603,6 +603,20 @@ set(INFER_SRC_LIST
     "analyzer/analyzer.cc"
 )
 
+if (ENABLE_OPEN_SRC)
+    file(MAKE_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}/msprofiler_fwk_object)
+    if(EXISTS ${ASCEND_RUNTIME_DIR}/libmsprofiler_fwk.a)
+        execute_process(
+            COMMAND ar x  ${ASCEND_RUNTIME_DIR}/libmsprofiler_fwk.a
+            WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}/msprofiler_fwk_object    
+        )
+        file(GLOB msprofiler_fwk ${CMAKE_CURRENT_BINARY_DIR}/msprofiler_fwk_object/*.o)
+    else()
+        file(GENERATE OUTPUT ${CMAKE_BINARY_DIR}/msprofiler_fwk.cc CONTENT "")
+        set(msprofiler_fwk ${CMAKE_BINARY_DIR}/msprofiler_fwk.cc)
+    endif()
+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>)