|
|
@@ -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); |
|
|
|