Browse Source

all in one -- runtime

pull/1893/head
wqtshg 4 years ago
parent
commit
8f025492b4
3 changed files with 14 additions and 3 deletions
  1. +2
    -1
      ge/CMakeLists.txt
  2. +2
    -1
      ge/offline/CMakeLists.txt
  3. +10
    -1
      ge/opskernel_manager/ops_kernel_builder_manager.cc

+ 2
- 1
ge/CMakeLists.txt View File

@@ -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


+ 2
- 1
ge/offline/CMakeLists.txt View File

@@ -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


+ 10
- 1
ge/opskernel_manager/ops_kernel_builder_manager.cc View File

@@ -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",


Loading…
Cancel
Save