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

!37951 [MS][LITE] fix converter c++ api convert method 

Merge pull request !37951 from luoyuan/fix-converter-convert-method
This commit is contained in:
i-robot 2022-07-13 06:01:05 +00:00 committed by Gitee
parent cf796583a9 12fd8795c2
commit 0135378514
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mindspore/lite/tools/converter/converter.cc
View File

@ -589,7 +589,14 @@ int RunConverter(const std::shared_ptr<ConverterPara> &param, void **model_data,
}
if (not_save) {
flatbuffers::FlatBufferBuilder builder(kFlatbuffersBuilderInitSize);
*model_data = static_cast<void *>(MetaGraphSerializer::GetMetaGraphPackedBuff(&builder, *meta_graph, data_size));
auto packed_buffer = MetaGraphSerializer::GetMetaGraphPackedBuff(&builder, *meta_graph, data_size);
auto buffer = malloc(*data_size);
if (buffer == nullptr) {
MS_LOG(ERROR) << "malloc failed.";
return RET_ERROR;
}
(void)memcpy_s(buffer, *data_size, packed_buffer, *data_size);
*model_data = buffer;
} else {
status = MetaGraphSerializer::Save(*meta_graph, param->output_file, encKey, keyLen, param->encrypt_mode);
}