mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

[MSLITE] change kMindIR_Opt to kMindIR_Lite

This commit is contained in:
ling 2022-02-15 16:33:24 +08:00
parent 08a10563c6
commit 1f0db7ac05
11 changed files with 26 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
include/api/types.h
View File

@ -39,7 +39,7 @@ enum ModelType : uint32_t {
kAIR = 1, kAIR = 1,
kOM = 2, kOM = 2,
kONNX = 3, kONNX = 3,
kMindIR_Opt = 4, kMindIR_Lite = 4,
// insert new data type here // insert new data type here
kUnknownType = 0xFFFFFFFF kUnknownType = 0xFFFFFFFF
}; };

4
mindspore/lite/java/src/main/native/model.cpp
View File

@ -74,7 +74,7 @@ extern "C" JNIEXPORT jlong JNICALL Java_com_mindspore_Model_buildByBuffer(JNIEnv
return reinterpret_cast<jlong>(nullptr); return reinterpret_cast<jlong>(nullptr);
} }
mindspore::ModelType c_model_type; mindspore::ModelType c_model_type;
if (model_type >= static_cast<int>(mindspore::kMindIR) && model_type <= static_cast<int>(mindspore::kMindIR_Opt)) { if (model_type >= static_cast<int>(mindspore::kMindIR) && model_type <= static_cast<int>(mindspore::kMindIR_Lite)) {
c_model_type = static_cast<mindspore::ModelType>(model_type); c_model_type = static_cast<mindspore::ModelType>(model_type);
} else { } else {
MS_LOGE("Invalid model type : %d", model_type); MS_LOGE("Invalid model type : %d", model_type);
@ -133,7 +133,7 @@ extern "C" JNIEXPORT jlong JNICALL Java_com_mindspore_Model_buildByPath(JNIEnv *
jcharArray key_str, jstring dec_mod) { jcharArray key_str, jstring dec_mod) {
auto c_model_path = env->GetStringUTFChars(model_path, JNI_FALSE); auto c_model_path = env->GetStringUTFChars(model_path, JNI_FALSE);
mindspore::ModelType c_model_type; mindspore::ModelType c_model_type;
if (model_type >= static_cast<int>(mindspore::kMindIR) && model_type <= static_cast<int>(mindspore::kMindIR_Opt)) { if (model_type >= static_cast<int>(mindspore::kMindIR) && model_type <= static_cast<int>(mindspore::kMindIR_Lite)) {
c_model_type = static_cast<mindspore::ModelType>(model_type); c_model_type = static_cast<mindspore::ModelType>(model_type);
} else { } else {
MS_LOGE("Invalid model type : %d", model_type); MS_LOGE("Invalid model type : %d", model_type);

6
mindspore/lite/src/cxx_api/serialization.cc
View File

@ -53,7 +53,7 @@ Status Serialization::Load(const void *model_data, size_t data_size, ModelType m
MS_LOG(ERROR) << "graph is nullptr."; MS_LOG(ERROR) << "graph is nullptr.";
return kLiteNullptr; return kLiteNullptr;
} }
if (model_type != kMindIR && model_type != kMindIR_Opt) { if (model_type != kMindIR && model_type != kMindIR_Lite) {
MS_LOG(ERROR) << "Unsupported IR."; MS_LOG(ERROR) << "Unsupported IR.";
return kLiteInputParamInvalid; return kLiteInputParamInvalid;
} }
@ -95,7 +95,7 @@ Status Serialization::Load(const std::vector<char> &file, ModelType model_type,
MS_LOG(ERROR) << "graph is nullptr."; MS_LOG(ERROR) << "graph is nullptr.";
return kLiteNullptr; return kLiteNullptr;
} }
if (model_type != kMindIR && model_type != kMindIR_Opt) { if (model_type != kMindIR && model_type != kMindIR_Lite) {
MS_LOG(ERROR) << "Unsupported IR."; MS_LOG(ERROR) << "Unsupported IR.";
return kLiteInputParamInvalid; return kLiteInputParamInvalid;
} }
@ -157,7 +157,7 @@ Status Serialization::ExportModel(const Model &model, ModelType model_type, cons
MS_LOG(ERROR) << "Model is not TrainModel."; MS_LOG(ERROR) << "Model is not TrainModel.";
return kLiteError; return kLiteError;
} }
if (model_type != kMindIR && model_type != kMindIR_Opt) { if (model_type != kMindIR && model_type != kMindIR_Lite) {
MS_LOG(ERROR) << "Unsupported Export Format " << model_type; MS_LOG(ERROR) << "Unsupported Export Format " << model_type;
return kLiteParamInvalid; return kLiteParamInvalid;
} }

16
mindspore/lite/src/lite_session.cc
View File

@ -1653,7 +1653,7 @@ session::LiteSession *session::LiteSession::CreateSession(const char *model_buf,
return nullptr; return nullptr;
} }
auto ret = reinterpret_cast<lite::LiteSession *>(session)->LoadModelAndCompileByBuf( auto ret = reinterpret_cast<lite::LiteSession *>(session)->LoadModelAndCompileByBuf(
model_buf, mindspore::ModelType::kMindIR_Opt, size); model_buf, mindspore::ModelType::kMindIR_Lite, size);
if (ret != RET_OK) { if (ret != RET_OK) {
MS_LOG(ERROR) << "Init session failed"; MS_LOG(ERROR) << "Init session failed";
delete session; delete session;
@ -1669,7 +1669,7 @@ session::LiteSession *lite::LiteSession::CreateSession(const std::string &model_
return nullptr; return nullptr;
} }
auto ret = reinterpret_cast<lite::LiteSession *>(session)->LoadModelAndCompileByPath( auto ret = reinterpret_cast<lite::LiteSession *>(session)->LoadModelAndCompileByPath(
model_path, mindspore::ModelType::kMindIR_Opt); model_path, mindspore::ModelType::kMindIR_Lite);
if (ret != RET_OK) { if (ret != RET_OK) {
MS_LOG(ERROR) << "Init session failed"; MS_LOG(ERROR) << "Init session failed";
delete session; delete session;
@ -1680,10 +1680,10 @@ session::LiteSession *lite::LiteSession::CreateSession(const std::string &model_
mindspore::ModelType lite::LiteSession::LoadModelByBuff(const char *model_buf, const size_t &buf_size, char **lite_buf, mindspore::ModelType lite::LiteSession::LoadModelByBuff(const char *model_buf, const size_t &buf_size, char **lite_buf,
size_t *size, mindspore::ModelType model_type) { size_t *size, mindspore::ModelType model_type) {
if (model_type == mindspore::ModelType::kMindIR_Opt) { if (model_type == mindspore::ModelType::kMindIR_Lite) {
*size = buf_size; *size = buf_size;
*lite_buf = const_cast<char *>(model_buf); *lite_buf = const_cast<char *>(model_buf);
return mindspore::ModelType::kMindIR_Opt; return mindspore::ModelType::kMindIR_Lite;
} }
if (model_type != mindspore::ModelType::kMindIR) { if (model_type != mindspore::ModelType::kMindIR) {
@ -1696,7 +1696,7 @@ mindspore::ModelType lite::LiteSession::LoadModelByBuff(const char *model_buf, c
MS_LOG(DEBUG) << "The kMindIR type model buffer is valid mslite model buffer"; MS_LOG(DEBUG) << "The kMindIR type model buffer is valid mslite model buffer";
*size = buf_size; *size = buf_size;
*lite_buf = const_cast<char *>(model_buf); *lite_buf = const_cast<char *>(model_buf);
return mindspore::ModelType::kMindIR_Opt; return mindspore::ModelType::kMindIR_Lite;
} }
MS_LOG(WARNING) << "Invalid mslite model."; MS_LOG(WARNING) << "Invalid mslite model.";
return mindspore::ModelType::kMindIR; return mindspore::ModelType::kMindIR;
@ -1705,10 +1705,10 @@ mindspore::ModelType lite::LiteSession::LoadModelByBuff(const char *model_buf, c
mindspore::ModelType lite::LiteSession::LoadModelByBuff(const char *model_buf, const size_t &buf_size, char **lite_buf, mindspore::ModelType lite::LiteSession::LoadModelByBuff(const char *model_buf, const size_t &buf_size, char **lite_buf,
size_t *size, mindspore::ModelType model_type, size_t *size, mindspore::ModelType model_type,
const std::shared_ptr<mindspore::Context> &ms_context) { const std::shared_ptr<mindspore::Context> &ms_context) {
if (model_type == mindspore::ModelType::kMindIR_Opt) { if (model_type == mindspore::ModelType::kMindIR_Lite) {
*size = buf_size; *size = buf_size;
*lite_buf = const_cast<char *>(model_buf); *lite_buf = const_cast<char *>(model_buf);
return mindspore::ModelType::kMindIR_Opt; return mindspore::ModelType::kMindIR_Lite;
} }
if (model_type != mindspore::ModelType::kMindIR) { if (model_type != mindspore::ModelType::kMindIR) {
@ -1721,7 +1721,7 @@ mindspore::ModelType lite::LiteSession::LoadModelByBuff(const char *model_buf, c
MS_LOG(DEBUG) << "The kMindIR type model buffer is valid mslite model buffer"; MS_LOG(DEBUG) << "The kMindIR type model buffer is valid mslite model buffer";
*size = buf_size; *size = buf_size;
*lite_buf = const_cast<char *>(model_buf); *lite_buf = const_cast<char *>(model_buf);
return mindspore::ModelType::kMindIR_Opt; return mindspore::ModelType::kMindIR_Lite;
} }
MS_LOG(WARNING) << "Invalid mslite model."; MS_LOG(WARNING) << "Invalid mslite model.";

2
mindspore/lite/test/st/mix_data_type_test.cc
View File

@ -204,7 +204,7 @@ TEST_F(MixDataTypeTest, mix1) {
auto status = impl->LoadConfig("MixDataTypeTestConfig"); auto status = impl->LoadConfig("MixDataTypeTestConfig");
ASSERT_EQ(status, kSuccess); ASSERT_EQ(status, kSuccess);
status = impl->Build(flat_model, size, kMindIR_Opt, context); status = impl->Build(flat_model, size, kMindIR_Lite, context);
ASSERT_EQ(status, kSuccess); ASSERT_EQ(status, kSuccess);
/* check */ /* check */

10
mindspore/lite/test/st/multiple_device_test.cc
View File

@ -363,7 +363,7 @@ TEST_F(MultipleDeviceTest, NewApi1) {
context->MutableDeviceInfo().push_back(std::make_shared<mindspore::GPUDeviceInfo>()); context->MutableDeviceInfo().push_back(std::make_shared<mindspore::GPUDeviceInfo>());
mindspore::Model *model = new mindspore::Model(); mindspore::Model *model = new mindspore::Model();
auto ret = model->Build(content, size, mindspore::kMindIR_Opt, context); auto ret = model->Build(content, size, mindspore::kMindIR_Lite, context);
ASSERT_EQ(false, ret.IsOk()); ASSERT_EQ(false, ret.IsOk());
delete model; delete model;
@ -422,7 +422,7 @@ TEST_F(MultipleDeviceTest, NewApi5) {
context->MutableDeviceInfo().push_back(std::make_shared<mindspore::CPUDeviceInfo>()); context->MutableDeviceInfo().push_back(std::make_shared<mindspore::CPUDeviceInfo>());
auto model_impl = std::make_shared<mindspore::ModelImpl>(); auto model_impl = std::make_shared<mindspore::ModelImpl>();
auto ret = model_impl->Build(content, size, mindspore::kMindIR_Opt, context); auto ret = model_impl->Build(content, size, mindspore::kMindIR_Lite, context);
ASSERT_EQ(mindspore::kSuccess, ret.StatusCode()); ASSERT_EQ(mindspore::kSuccess, ret.StatusCode());
CheckResult(reinterpret_cast<const mindspore::lite::LiteSession *>(model_impl->GetSession())->get_kernels(), CheckResult(reinterpret_cast<const mindspore::lite::LiteSession *>(model_impl->GetSession())->get_kernels(),
@ -467,7 +467,7 @@ TEST_F(MultipleDeviceTest, NewApi6) {
context->MutableDeviceInfo().push_back(std::make_shared<mindspore::GPUDeviceInfo>()); context->MutableDeviceInfo().push_back(std::make_shared<mindspore::GPUDeviceInfo>());
auto model_impl = std::make_shared<mindspore::ModelImpl>(); auto model_impl = std::make_shared<mindspore::ModelImpl>();
auto ret = model_impl->Build(content, size, mindspore::kMindIR_Opt, context); auto ret = model_impl->Build(content, size, mindspore::kMindIR_Lite, context);
ASSERT_EQ(mindspore::kSuccess, ret.StatusCode()); ASSERT_EQ(mindspore::kSuccess, ret.StatusCode());
CheckResult(reinterpret_cast<const mindspore::lite::LiteSession *>(model_impl->GetSession())->get_kernels(), CheckResult(reinterpret_cast<const mindspore::lite::LiteSession *>(model_impl->GetSession())->get_kernels(),
@ -511,7 +511,7 @@ TEST_F(MultipleDeviceTest, NewApi7) {
context->MutableDeviceInfo().push_back(std::make_shared<mindspore::GPUDeviceInfo>()); context->MutableDeviceInfo().push_back(std::make_shared<mindspore::GPUDeviceInfo>());
auto model_impl = std::make_shared<mindspore::ModelImpl>(); auto model_impl = std::make_shared<mindspore::ModelImpl>();
auto ret = model_impl->Build(content, size, mindspore::kMindIR_Opt, context); auto ret = model_impl->Build(content, size, mindspore::kMindIR_Lite, context);
ASSERT_EQ(mindspore::kSuccess, ret.StatusCode()); ASSERT_EQ(mindspore::kSuccess, ret.StatusCode());
CheckResult(reinterpret_cast<const mindspore::lite::LiteSession *>(model_impl->GetSession())->get_kernels(), CheckResult(reinterpret_cast<const mindspore::lite::LiteSession *>(model_impl->GetSession())->get_kernels(),
@ -536,7 +536,7 @@ TEST_F(MultipleDeviceTest, NewApi8) {
context->MutableDeviceInfo().push_back(std::make_shared<mindspore::KirinNPUDeviceInfo>()); context->MutableDeviceInfo().push_back(std::make_shared<mindspore::KirinNPUDeviceInfo>());
auto model_impl = std::make_shared<mindspore::ModelImpl>(); auto model_impl = std::make_shared<mindspore::ModelImpl>();
auto ret = model_impl->Build(content, size, mindspore::kMindIR_Opt, context); auto ret = model_impl->Build(content, size, mindspore::kMindIR_Lite, context);
ASSERT_EQ(mindspore::kSuccess, ret.StatusCode()); ASSERT_EQ(mindspore::kSuccess, ret.StatusCode());
CheckResult(reinterpret_cast<const mindspore::lite::LiteSession *>(model_impl->GetSession())->get_kernels(), CheckResult(reinterpret_cast<const mindspore::lite::LiteSession *>(model_impl->GetSession())->get_kernels(),

2
mindspore/lite/test/ut/src/registry/registry_custom_op_test.cc
View File

@ -206,7 +206,7 @@ TEST_F(TestRegistryCustomOp, TestCustomAdd) {
// build a model // build a model
auto model = std::make_shared<mindspore::Model>(); auto model = std::make_shared<mindspore::Model>();
auto ret = model->Build(content, size, kMindIR_Opt, context); auto ret = model->Build(content, size, kMindIR_Lite, context);
ASSERT_EQ(kSuccess, ret.StatusCode()); ASSERT_EQ(kSuccess, ret.StatusCode());
auto inputs = model->GetInputs(); auto inputs = model->GetInputs();
ASSERT_EQ(inputs.size(), 2); ASSERT_EQ(inputs.size(), 2);

2
mindspore/lite/test/ut/src/registry/registry_gpu_custom_op_test.cc
View File

@ -515,7 +515,7 @@ TEST_F(TestGPURegistryCustomOp, TestGPUCustomAdd) {
// build a model // build a model
auto model = std::make_shared<mindspore::Model>(); auto model = std::make_shared<mindspore::Model>();
auto ret = model->Build(content, size, kMindIR_Opt, context); auto ret = model->Build(content, size, kMindIR_Lite, context);
ASSERT_EQ(kSuccess, ret.StatusCode()); ASSERT_EQ(kSuccess, ret.StatusCode());
auto inputs = model->GetInputs(); auto inputs = model->GetInputs();
ASSERT_EQ(inputs.size(), 2); ASSERT_EQ(inputs.size(), 2);

2
mindspore/lite/test/ut/src/registry/registry_test.cc
View File

@ -166,7 +166,7 @@ TEST_F(TestRegistry, TestAdd) {
// build a model // build a model
auto model = std::make_shared<mindspore::Model>(); auto model = std::make_shared<mindspore::Model>();
auto ret = model->Build(content, size, kMindIR_Opt, context); auto ret = model->Build(content, size, kMindIR_Lite, context);
ASSERT_EQ(kSuccess, ret.StatusCode()); ASSERT_EQ(kSuccess, ret.StatusCode());
auto inputs = model->GetInputs(); auto inputs = model->GetInputs();
ASSERT_EQ(inputs.size(), 2); ASSERT_EQ(inputs.size(), 2);

2
mindspore/lite/test/ut/src/runtime/runtime_convert_tests.cc
View File

@ -53,7 +53,7 @@ TEST_F(RuntimeConvert, relu2) {
Model model; Model model;
auto context = std::make_shared<mindspore::Context>(); auto context = std::make_shared<mindspore::Context>();
context->MutableDeviceInfo().push_back(std::make_shared<mindspore::CPUDeviceInfo>()); context->MutableDeviceInfo().push_back(std::make_shared<mindspore::CPUDeviceInfo>());
Status build_ret = model.Build("./relu.mindir", mindspore::kMindIR_Opt, context); Status build_ret = model.Build("./relu.mindir", mindspore::kMindIR_Lite, context);
ASSERT_NE(build_ret, Status::OK()); ASSERT_NE(build_ret, Status::OK());
} }

4
mindspore/lite/tools/benchmark/benchmark_base.h
View File

@ -74,7 +74,7 @@ extern const std::unordered_map<int, std::string> kTypeIdMap;
extern const std::unordered_map<mindspore::Format, std::string> kTensorFormatMap; extern const std::unordered_map<mindspore::Format, std::string> kTensorFormatMap;
const std::unordered_map<std::string, mindspore::ModelType> ModelTypeMap{ const std::unordered_map<std::string, mindspore::ModelType> ModelTypeMap{
{"MindIR_Opt", mindspore::ModelType::kMindIR_Opt}, {"MindIR", mindspore::ModelType::kMindIR}}; {"MindIR_Lite", mindspore::ModelType::kMindIR_Lite}, {"MindIR", mindspore::ModelType::kMindIR}};
namespace dump { namespace dump {
constexpr auto kConfigPath = "MINDSPORE_DUMP_CONFIG"; constexpr auto kConfigPath = "MINDSPORE_DUMP_CONFIG";
@ -116,7 +116,7 @@ class MS_API BenchmarkFlags : public virtual FlagParser {
BenchmarkFlags() { BenchmarkFlags() {
// common // common
AddFlag(&BenchmarkFlags::model_file_, "modelFile", "Input model file", ""); AddFlag(&BenchmarkFlags::model_file_, "modelFile", "Input model file", "");
AddFlag(&BenchmarkFlags::model_type_, "modelType", "Input model type. MindIR | MindIR_Opt", "MindIR"); AddFlag(&BenchmarkFlags::model_type_, "modelType", "Input model type. MindIR | MindIR_Lite", "MindIR");
AddFlag(&BenchmarkFlags::in_data_file_, "inDataFile", "Input data file, if not set, use random input", ""); AddFlag(&BenchmarkFlags::in_data_file_, "inDataFile", "Input data file, if not set, use random input", "");
AddFlag(&BenchmarkFlags::config_file_, "configFile", "Config file", ""); AddFlag(&BenchmarkFlags::config_file_, "configFile", "Config file", "");
AddFlag(&BenchmarkFlags::device_, "device", "CPU | GPU | NPU | Ascend310 | Ascend710", "CPU"); AddFlag(&BenchmarkFlags::device_, "device", "CPU | GPU | NPU | Ascend310 | Ascend710", "CPU");