forked from mindspore-Ecosystem/mindspore
!40139 [MS][LITE] cloud inference pkg
Merge pull request !40139 from sunsuodong/cloud_inference_pkg2
This commit is contained in:
commit
839a7fae7c
|
@ -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)
|
||||
|
|
|
@ -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")
|
||||
|
|
|
@ -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;
|
||||
|
|
|
@ -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;
|
||||
|
|
Loading…
Reference in New Issue