From 31a31004ff1b1a8032e791dd55eb9d672ea812fe Mon Sep 17 00:00:00 2001 From: zhupuxu Date: Sat, 8 May 2021 14:19:43 +0800 Subject: [PATCH] fix bug for infer failed at 310 Signed-off-by: zhupuxu --- mindspore/ccsrc/cxx_api/graph/graph_data.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/ccsrc/cxx_api/graph/graph_data.cc b/mindspore/ccsrc/cxx_api/graph/graph_data.cc index 85be64f8ff..7dcef50182 100644 --- a/mindspore/ccsrc/cxx_api/graph/graph_data.cc +++ b/mindspore/ccsrc/cxx_api/graph/graph_data.cc @@ -41,7 +41,7 @@ Graph::GraphData::GraphData(Buffer om_data, enum ModelType model_type) ge::ModelData model_data; model_data.model_data = om_data.MutableData(); model_data.model_len = om_data.DataSize(); - ge::Status ret = helper.LoadModel(model_data); + ge::Status ret = helper.LoadRootModel(model_data); if (ret != ge::SUCCESS) { MS_LOG(EXCEPTION) << "Invalid input data cannot parse to om."; }