!40139 [MS][LITE] cloud inference pkg

Merge pull request !40139 from sunsuodong/cloud_inference_pkg2
This commit is contained in:
i-robot 2022-08-10 02:08:45 +00:00 committed by Gitee
commit 839a7fae7c
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
4 changed files with 5 additions and 3 deletions

View File

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

View File

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

View File

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

View File

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