|
|
@@ -434,27 +434,10 @@ Status GraphExecutor::ExecuteGraphWithStream(GraphId graph_id, |
|
|
|
|
|
|
|
GE_CHECK_NOTNULL_EXEC(ge_root_model, return FAILED); |
|
|
|
auto model_id = ge_root_model->GetModelId(); |
|
|
|
InputData input_data; |
|
|
|
input_data.index = 0; |
|
|
|
input_data.model_id = model_id; |
|
|
|
std::vector<GeTensorDesc> input_desc; |
|
|
|
auto ret = GetExecuteData(input_tensor, input_data.blobs, input_desc); |
|
|
|
if (ret != SUCCESS) { |
|
|
|
return ret; |
|
|
|
} |
|
|
|
OutputData output_data; |
|
|
|
output_data.index = 0; |
|
|
|
output_data.model_id = model_id; |
|
|
|
std::vector<GeTensorDesc> output_desc; |
|
|
|
ret = GetExecuteData(output_tensor, output_data.blobs, output_desc); |
|
|
|
if (ret != SUCCESS) { |
|
|
|
return ret; |
|
|
|
} |
|
|
|
|
|
|
|
auto async_mode = true; |
|
|
|
auto model_manager = ge::ModelManager::GetInstance(); |
|
|
|
GE_CHECK_NOTNULL(model_manager); |
|
|
|
ret = model_manager->ExecuteModel(model_id, stream, async_mode, input_tensor, output_tensor); |
|
|
|
auto ret = model_manager->ExecuteModel(model_id, stream, async_mode, input_tensor, output_tensor); |
|
|
|
if (ret != SUCCESS) { |
|
|
|
return ret; |
|
|
|
} |
|
|
|