Browse Source

!3118 add tbe build json with full name for debug

Merge pull request !3118 from jjfeing/master
tags/v0.6.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
536f7533e8
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_build.cc

+ 2
- 0
mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_build.cc View File

@@ -43,6 +43,7 @@ constexpr auto kJInputs = "inputs";
constexpr auto kJOutputs = "outputs";
constexpr auto kJAttrs = "attrs";
constexpr auto kJKernelName = "kernel_name";
constexpr auto kJFullName = "full_name";
constexpr auto kJOpInfo = "op_info";
constexpr auto kJDtype = "dtype";
constexpr auto kJtype = "type";
@@ -125,6 +126,7 @@ bool TbeKernelJsonCreator::GenTbeSingleKernelJson(const std::shared_ptr<mindspor
op_info_json[kJKernelName] = json_name_;
}
(*kernel_json)[kJOpInfo] = op_info_json;
(*kernel_json)[kJFullName] = anf_node->fullname_with_scope();
if (creater_type_ == SINGLE_BUILD) {
TbeUtils::SaveJsonInfo(json_name_, json_info_);
}


Loading…
Cancel
Save