From dfd894c19093e3fa878ff2b46e209fe1c1dbdc22 Mon Sep 17 00:00:00 2001 From: jjfeing Date: Mon, 30 Nov 2020 17:26:50 +0800 Subject: [PATCH] add trace in backend ascend --- mindspore/ccsrc/backend/kernel_compiler/kernel_query.cc | 3 ++- .../ccsrc/backend/kernel_compiler/tbe/tbe_kernel_build.cc | 4 +++- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/mindspore/ccsrc/backend/kernel_compiler/kernel_query.cc b/mindspore/ccsrc/backend/kernel_compiler/kernel_query.cc index 268283b3f6..844e870765 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/kernel_query.cc +++ b/mindspore/ccsrc/backend/kernel_compiler/kernel_query.cc @@ -25,6 +25,7 @@ #include "backend/kernel_compiler/akg/akg_kernel_metadata.h" #include "backend/session/anf_runtime_algorithm.h" #include "utils/ms_context.h" +#include "utils/trace_base.h" namespace mindspore { namespace kernel { @@ -94,7 +95,7 @@ void KernelQueryAll(const CNodePtr &kernel_node, MS_EXCEPTION(NotExistsError) << "Failed to obtain operator info, Please check whether the operator info is registered, Op full name:" << kernel_node->fullname_with_scope() << "Node Type: " << op_name - << ", Node DebugString: " << kernel_node->DebugString(); + << ", Node DebugString: " << kernel_node->DebugString() << "\n trace: " << trace::DumpSourceLines(kernel_node); } } diff --git a/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_build.cc b/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_build.cc index f9642dd1f5..1f844c72c4 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_build.cc +++ b/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_build.cc @@ -27,6 +27,7 @@ #include "backend/kernel_compiler/tbe/tbe_utils.h" #include "utils/ms_context.h" #include "runtime/dev.h" +#include "utils/trace_base.h" namespace mindspore { namespace kernel { @@ -228,7 +229,8 @@ bool TbeKernelJsonCreator::GenInputList(const std::shared_ptr &anf_node input_list->emplace_back(input_desc_json); continue; } - MS_LOG(ERROR) << "Input num: " << *real_input_index << " is not match op inputs"; + MS_LOG(ERROR) << "Input num: " << *real_input_index << " is not match op inputs." + << "\n trace:" << trace::DumpSourceLines(anf_node); return false; } if (op_name == "BatchNorm") {