!9228 add trace in tbe select & build

From: @jjfeing
Reviewed-by: @kisnwang,@chujinjin
Signed-off-by: @chujinjin
pull/9228/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 5ebfba6f03

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

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

Loading…
Cancel
Save