From 8f025492b42b14263c78e21d07a5b71aa0318f57 Mon Sep 17 00:00:00 2001
From: wqtshg <wangtao123@huawei.com>
Date: Mon, 28 Jun 2021 21:13:58 +0800
Subject: [PATCH] all in one -- runtime

---
 ge/CMakeLists.txt                                  |  3 ++-
 ge/offline/CMakeLists.txt                          |  3 ++-
 ge/opskernel_manager/ops_kernel_builder_manager.cc | 11 ++++++++++-
 3 files changed, 14 insertions(+), 3 deletions(-)

diff --git a/ge/CMakeLists.txt b/ge/CMakeLists.txt
index dc80597c..1e6861e4 100755
--- a/ge/CMakeLists.txt
+++ b/ge/CMakeLists.txt
@@ -866,7 +866,8 @@ target_link_libraries(ge_compiler PRIVATE
     c_sec
     error_manager
     slog
-    runtime_compile
+    $<$<NOT:$<BOOL:${ENABLE_OPEN_SRC}>>:$<BUILD_INTERFACE:runtime>>
+    $<$<BOOL:${ENABLE_OPEN_SRC}>:$<BUILD_INTERFACE:runtime_compile>>
     -Wl,--as-needed
     json
     -lrt
diff --git a/ge/offline/CMakeLists.txt b/ge/offline/CMakeLists.txt
index 3a320226..e8f115f7 100644
--- a/ge/offline/CMakeLists.txt
+++ b/ge/offline/CMakeLists.txt
@@ -58,11 +58,12 @@ target_link_libraries(atc_atc.bin PRIVATE
     c_sec
     graph
     error_manager
+    $<$<NOT:$<BOOL:${ENABLE_OPEN_SRC}>>:$<BUILD_INTERFACE:runtime>>
+    $<$<BOOL:${ENABLE_OPEN_SRC}>:$<BUILD_INTERFACE:runtime_compile>>
     ge_compiler
     parser_common
     gflags
     json
-    runtime_compile
     slog
     static_mmpa
     -lrt
diff --git a/ge/opskernel_manager/ops_kernel_builder_manager.cc b/ge/opskernel_manager/ops_kernel_builder_manager.cc
index 33ffddf5..fe6edb6a 100644
--- a/ge/opskernel_manager/ops_kernel_builder_manager.cc
+++ b/ge/opskernel_manager/ops_kernel_builder_manager.cc
@@ -20,6 +20,7 @@
 
 namespace ge {
 namespace {
+#ifdef ENABLE_OPEN_SRC
 const std::vector<std::string> kBasicBuilderLibs = {
     "libge_local_opskernel_builder.so",
     "libhost_cpu_opskernel_builder.so",
@@ -27,7 +28,15 @@ const std::vector<std::string> kBasicBuilderLibs = {
     "libaicpu_ascend_builder.so",
     "libaicpu_tf_builder.so"
 };
-
+#else
+const std::vector<std::string> kBasicBuilderLibs = {
+     "libge_local_opskernel_builder.so",
+     "libhost_cpu_opskernel_builder.so",
+     "librts_engine.so",
+     "libaicpu_ascend_engine.so",
+     "libaicpu_tf_engine.so"
+};
+#endif
 const std::vector<std::string> kHcclBuilderLibs = {
     "libhcom_opskernel_builder.so",
     "libhvd_opskernel_builder.so",