xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!18737 change nnie_converter so relative path 

Merge pull request !18737 from zhaodezan/master
This commit is contained in:
i-robot 2021-06-23 03:03:59 +00:00 committed by Gitee
parent fcfa6626c8 c6b633730d
commit 1f2e87a0f5
5 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
cmake/package_lite.cmake
View File

@ -398,7 +398,9 @@ else()
file(GLOB PROTOBUF_LIB_PATH ${TOP_DIR}/mindspore/lite/build/.mslib/protobuf_*/lib/libprotobuf.a)
install(FILES ${PROTOBUF_LIB_PATH} DESTINATION ${CONVERTER_ROOT_DIR}/lib
COMPONENT ${RUNTIME_COMPONENT_NAME})
install(FILES ${TOP_DIR}/mindspore/lite/tools/providers/NNIE/3516D/libmslite_converter_plugin.so
install(FILES ${TOP_DIR}/mindspore/lite/tools/providers/NNIE/converter.cfg
DESTINATION ${CONVERTER_ROOT_DIR}/ COMPONENT ${RUNTIME_COMPONENT_NAME})
install(FILES ${TOP_DIR}/mindspore/lite/tools/providers/NNIE/3516D/libmslite_nnie_converter.so
DESTINATION ${CONVERTER_ROOT_DIR}/providers/3516D/ COMPONENT ${RUNTIME_COMPONENT_NAME})
install(FILES ${TOP_DIR}/mindspore/lite/tools/providers/NNIE/3516D/libmslite_nnie_data_process.so
DESTINATION ${CONVERTER_ROOT_DIR}/providers/3516D COMPONENT ${RUNTIME_COMPONENT_NAME})

6
mindspore/lite/tools/converter/converter_flags.cc
View File

@ -23,6 +23,7 @@
#include <vector>
#include <memory>
#include "ir/dtype/type_id.h"
#include "common/file_utils.h"
namespace mindspore {
namespace lite {
@ -198,7 +199,10 @@ int Flags::InitConfigFile() {
auto plugins_path_str = GetStrFromConfigFile(this->configFile, "plugin_path");
if (!plugins_path_str.empty()) {
const char *delimiter = ";";
this->pluginsPath = SplitStringToVector(plugins_path_str, *delimiter);
auto relative_path = SplitStringToVector(plugins_path_str, *delimiter);
for (size_t i = 0; i < relative_path.size(); i++) {
this->pluginsPath.push_back(RealPath(relative_path[i].c_str()));
}
}
auto disable_fusion_flag = GetStrFromConfigFile(this->configFile, "disable_fusion");

BIN
mindspore/lite/tools/providers/NNIE/3516D/libmslite_converter_plugin.so
View File

Binary file not shown.

BIN
mindspore/lite/tools/providers/NNIE/3516D/libmslite_nnie_converter.so Executable file
View File

Binary file not shown.

1
mindspore/lite/tools/providers/NNIE/converter.cfg Normal file
View File

@ -0,0 +1 @@
plugin_path=./providers/3516D/libmslite_nnie_converter.so