Browse Source

GPU add log in loadinputdata when tensor input != graph input

tags/v1.0.0
VectorSL 5 years ago
parent
commit
f36f72b99b
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      mindspore/ccsrc/backend/session/gpu_session.cc

+ 3
- 1
mindspore/ccsrc/backend/session/gpu_session.cc View File

@@ -136,7 +136,9 @@ void GPUSession::LoadInputData(const std::shared_ptr<KernelGraph> &kernel_graph,
auto input_nodes = kernel_graph->inputs(); auto input_nodes = kernel_graph->inputs();
auto ms_context = MsContext::GetInstance(); auto ms_context = MsContext::GetInstance();
MS_EXCEPTION_IF_NULL(ms_context); MS_EXCEPTION_IF_NULL(ms_context);

if (inputs.size() != input_nodes.size()) {
MS_LOG(EXCEPTION) << "Tensor input:" << inputs.size() << " is not equal graph inputs:" << input_nodes.size();
}
for (size_t i = 0; i < inputs.size(); ++i) { for (size_t i = 0; i < inputs.size(); ++i) {
auto tensor = inputs[i]; auto tensor = inputs[i];
MS_EXCEPTION_IF_NULL(tensor); MS_EXCEPTION_IF_NULL(tensor);


Loading…
Cancel
Save