diff --git a/mindspore/ccsrc/backend/session/gpu_session.cc b/mindspore/ccsrc/backend/session/gpu_session.cc index cf2508216b..d4475da1c3 100644 --- a/mindspore/ccsrc/backend/session/gpu_session.cc +++ b/mindspore/ccsrc/backend/session/gpu_session.cc @@ -136,7 +136,9 @@ void GPUSession::LoadInputData(const std::shared_ptr &kernel_graph, auto input_nodes = kernel_graph->inputs(); auto ms_context = MsContext::GetInstance(); 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) { auto tensor = inputs[i]; MS_EXCEPTION_IF_NULL(tensor);