Browse Source

!9228 add trace in tbe select & build

From: @jjfeing
Reviewed-by: @kisnwang,@chujinjin
Signed-off-by: @chujinjin
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
5ebfba6f03
2 changed files with 5 additions and 2 deletions
  1. +2
    -1
      mindspore/ccsrc/backend/kernel_compiler/kernel_query.cc
  2. +3
    -1
      mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_build.cc

+ 2
- 1
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/kernel_compiler/akg/akg_kernel_metadata.h"
#include "backend/session/anf_runtime_algorithm.h" #include "backend/session/anf_runtime_algorithm.h"
#include "utils/ms_context.h" #include "utils/ms_context.h"
#include "utils/trace_base.h"


namespace mindspore { namespace mindspore {
namespace kernel { namespace kernel {
@@ -94,7 +95,7 @@ void KernelQueryAll(const CNodePtr &kernel_node,
MS_EXCEPTION(NotExistsError) MS_EXCEPTION(NotExistsError)
<< "Failed to obtain operator info, Please check whether the operator info is registered, Op full name:" << "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 << 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);
} }
} }




+ 3
- 1
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 "backend/kernel_compiler/tbe/tbe_utils.h"
#include "utils/ms_context.h" #include "utils/ms_context.h"
#include "runtime/dev.h" #include "runtime/dev.h"
#include "utils/trace_base.h"


namespace mindspore { namespace mindspore {
namespace kernel { namespace kernel {
@@ -228,7 +229,8 @@ bool TbeKernelJsonCreator::GenInputList(const std::shared_ptr<AnfNode> &anf_node
input_list->emplace_back(input_desc_json); input_list->emplace_back(input_desc_json);
continue; 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; return false;
} }
if (op_name == "BatchNorm") { if (op_name == "BatchNorm") {


Loading…
Cancel
Save