From 820ef77d133254af558ef81a3c2d07e808b63489 Mon Sep 17 00:00:00 2001 From: sunsuodong Date: Mon, 8 Aug 2022 01:16:09 -0700 Subject: [PATCH] cloud inference pkg --- cmake/package_lite.cmake | 2 +- mindspore/lite/src/extendrt/CMakeLists.txt | 2 ++ .../src/extendrt/delegate/plugin/tensorrt_executor_plugin.cc | 2 +- .../src/extendrt/kernel/ascend/plugin/ascend_kernel_plugin.cc | 2 +- 4 files changed, 5 insertions(+), 3 deletions(-) diff --git a/cmake/package_lite.cmake b/cmake/package_lite.cmake index aded3608d5f..5164d4b23cf 100644 --- a/cmake/package_lite.cmake +++ b/cmake/package_lite.cmake @@ -21,7 +21,7 @@ set(TURBO_DIR ${RUNTIME_PKG_NAME}/runtime/third_party/libjpeg-turbo) set(GLOG_DIR ${RUNTIME_PKG_NAME}/runtime/third_party/glog) set(SECUREC_DIR ${RUNTIME_PKG_NAME}/runtime/third_party/securec) set(MINDSPORE_LITE_LIB_NAME libmindspore-lite) -set(MINDSPORE_LITE_EXTENDRT_LIB_NAME libmindspore-extendrt) +set(MINDSPORE_LITE_EXTENDRT_LIB_NAME libmindspore-lite) set(MINDSPORE_CORE_LIB_NAME libmindspore_core) set(BENCHMARK_NAME benchmark) set(MSLITE_NNIE_LIB_NAME libmslite_nnie) diff --git a/mindspore/lite/src/extendrt/CMakeLists.txt b/mindspore/lite/src/extendrt/CMakeLists.txt index 579629efea5..d6c0cbb06e4 100644 --- a/mindspore/lite/src/extendrt/CMakeLists.txt +++ b/mindspore/lite/src/extendrt/CMakeLists.txt @@ -201,3 +201,5 @@ else() add_library(mindspore-extendrt OBJECT ${MSLITE_EXTEND_RUNTIME_SRC}) add_dependencies(mindspore-extendrt fbs_inner_src) endif() + +set_target_properties(mindspore-extendrt PROPERTIES OUTPUT_NAME "mindspore-lite") diff --git a/mindspore/lite/src/extendrt/delegate/plugin/tensorrt_executor_plugin.cc b/mindspore/lite/src/extendrt/delegate/plugin/tensorrt_executor_plugin.cc index a28c397d26f..e4142541597 100644 --- a/mindspore/lite/src/extendrt/delegate/plugin/tensorrt_executor_plugin.cc +++ b/mindspore/lite/src/extendrt/delegate/plugin/tensorrt_executor_plugin.cc @@ -45,7 +45,7 @@ bool TensorRTPlugin::Register() { return true; } std::string plugin_path; - auto ret = DLSoPath("libmindspore-extendrt.so", kTensorRtPluginSoName, &plugin_path); + auto ret = DLSoPath("libmindspore-lite.so", kTensorRtPluginSoName, &plugin_path); if (ret != kSuccess) { MS_LOG(ERROR) << "Get real path of " << kTensorRtPluginSoName << " failed."; return false; diff --git a/mindspore/lite/src/extendrt/kernel/ascend/plugin/ascend_kernel_plugin.cc b/mindspore/lite/src/extendrt/kernel/ascend/plugin/ascend_kernel_plugin.cc index 8bd4ceee75c..a0f942552f8 100644 --- a/mindspore/lite/src/extendrt/kernel/ascend/plugin/ascend_kernel_plugin.cc +++ b/mindspore/lite/src/extendrt/kernel/ascend/plugin/ascend_kernel_plugin.cc @@ -38,7 +38,7 @@ void AscendKernelPlugin::Register() { return; } std::string ascend_kernel_plugin_path; - auto ret = DLSoPath("libmindspore-extendrt.so", "libascend_kernel_plugin.so", &ascend_kernel_plugin_path); + auto ret = DLSoPath("libmindspore-lite.so", "libascend_kernel_plugin.so", &ascend_kernel_plugin_path); if (ret != kSuccess) { MS_LOG(ERROR) << "Get real path of libascend_kernel_plugin.so failed."; return;