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

fix helper bug

This commit is contained in:
zhaoyingzhuo 2023-01-05 16:41:14 +08:00
parent 500a0d9c23
commit 86a2d96c73
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mindspore/lite/src/extendrt/cxx_api/model/model_impl.cc
View File

@ -155,6 +155,15 @@ Status ModelImpl::Build(const std::string &model_path, ModelType model_type,
Status ModelImpl::ConvertGraphOnline(const FuncGraphPtr &func_graph, const std::shared_ptr<Context> &model_context) {
MS_ASSERT(func_graph != nullptr);
auto device_list = model_context->MutableDeviceInfo();
for (const auto &device_info : device_list) {
if (device_info == nullptr) {
continue;
}
if (device_info->GetDeviceType() == DeviceType::kAscend && device_info->GetProvider() == "ge") {
return kSuccess;
}
}
auto value = func_graph->get_attr(lite::kIsOptimized);
if (value != nullptr) {
if (GetValue<bool>(value)) {