From 36fec5dca778b38f426ee8fa8a34fd51cc97ceb0 Mon Sep 17 00:00:00 2001 From: yao_yf Date: Fri, 20 Nov 2020 09:34:08 +0800 Subject: [PATCH] fix_assign_memory_bug_with_tuple_get_item_and_communication_ops --- mindspore/ccsrc/runtime/device/kernel_runtime.cc | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/mindspore/ccsrc/runtime/device/kernel_runtime.cc b/mindspore/ccsrc/runtime/device/kernel_runtime.cc index 2f09dab5bc..701c6f2cf4 100644 --- a/mindspore/ccsrc/runtime/device/kernel_runtime.cc +++ b/mindspore/ccsrc/runtime/device/kernel_runtime.cc @@ -477,12 +477,11 @@ DeviceAddressPtr KernelRuntime::PreAssignCNodeMemory(const AnfNodePtr &anf_node, auto cnode = anf_node->cast(); if (opt::IsNopNode(cnode)) { size_t kNopNodeInputSize = 2; - size_t kNopNodeRealInputIndex = 1; if (cnode->size() != kNopNodeInputSize) { MS_LOG(EXCEPTION) << cnode->fullname_with_scope() << " has invalid input size: " << cnode->size(); } auto input_node_with_index = AnfAlgo::GetPrevNodeOutput(anf_node, index); - return PreAssignCNodeMemory(cnode->input(kNopNodeRealInputIndex), input_node_with_index.second); + return PreAssignCNodeMemory(input_node_with_index.first, input_node_with_index.second); } auto kernel_mod = AnfAlgo::GetKernelMod(anf_node); MS_EXCEPTION_IF_NULL(kernel_mod);