From b4cd7860794fe12743e839120f546a7fa57e1d34 Mon Sep 17 00:00:00 2001 From: Adel Shafiei Date: Tue, 22 Dec 2020 10:50:26 -0500 Subject: [PATCH] fixed the check on output address --- mindspore/ccsrc/debug/debugger/debugger.cc | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/mindspore/ccsrc/debug/debugger/debugger.cc b/mindspore/ccsrc/debug/debugger/debugger.cc index 93efc12513..8004b667fb 100644 --- a/mindspore/ccsrc/debug/debugger/debugger.cc +++ b/mindspore/ccsrc/debug/debugger/debugger.cc @@ -1178,11 +1178,8 @@ void Debugger::LoadGraphOutputs() { } } for (size_t j = 0; j < output_size; ++j) { - auto kernel_info = static_cast(node->kernel_info()); - MS_EXCEPTION_IF_NULL(kernel_info); - auto addr_test = kernel_info->GetOutputAddr(j); - if (addr_test == nullptr) { - MS_LOG(INFO) << "Cannot find output addr for slot " << j << " for " << kernel_name; + if (!AnfAlgo::OutputAddrExist(node, j)) { + MS_LOG(INFO) << "Cannot find output addr for slot " << j << " for " << node->fullname_with_scope(); continue; } auto addr = AnfAlgo::GetOutputAddr(node, j);