From e647f71dd24e676aa14273904e5a090ae817502f Mon Sep 17 00:00:00 2001 From: wqtshg Date: Tue, 29 Jun 2021 09:08:08 +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 4a296e87..3fb63e62 100755 --- a/ge/CMakeLists.txt +++ b/ge/CMakeLists.txt @@ -873,7 +873,8 @@ target_link_libraries(ge_compiler PRIVATE c_sec error_manager slog - runtime_compile + $<$>:$> + $<$:$> opt_feature -Wl,--as-needed json diff --git a/ge/offline/CMakeLists.txt b/ge/offline/CMakeLists.txt index 3a320226..e11e4a03 100644 --- a/ge/offline/CMakeLists.txt +++ b/ge/offline/CMakeLists.txt @@ -62,7 +62,8 @@ target_link_libraries(atc_atc.bin PRIVATE 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..b8d28f62 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 ONLY_COMPILE_OPEN_SRC const std::vector kBasicBuilderLibs = { "libge_local_opskernel_builder.so", "libhost_cpu_opskernel_builder.so", @@ -27,7 +28,15 @@ const std::vector kBasicBuilderLibs = { "libaicpu_ascend_builder.so", "libaicpu_tf_builder.so" }; - +#else +const std::vector 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 kHcclBuilderLibs = { "libhcom_opskernel_builder.so", "libhvd_opskernel_builder.so",