xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!9228 add trace in tbe select & build 

From: @jjfeing
Reviewed-by: @kisnwang,@chujinjin
Signed-off-by: @chujinjin
This commit is contained in:
mindspore-ci-bot 2020-12-01 09:07:23 +08:00 committed by Gitee
parent ea191ed5da dfd894c190
commit 5ebfba6f03
2 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/ccsrc/backend/kernel_compiler/kernel_query.cc
View File

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

4
mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_build.cc
View File

@ -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<AnfNode> &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") {