Browse Source

!5233 GPU add log in LoadInputData

Merge pull request !5233 from VectorSL/r0.7
tags/v0.7.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
71dd8a4a71
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 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);


Loading…
Cancel
Save