Browse Source

!8025 [GraphKernel] Limited the length of fuse-op's name.

Merge pull request !8025 from TronZhang/limit_fuseop_name_len
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
c65f2f198d
2 changed files with 11 additions and 3 deletions
  1. +10
    -3
      mindspore/ccsrc/backend/kernel_compiler/akg/akg_kernel_json_generator.cc
  2. +1
    -0
      mindspore/ccsrc/backend/kernel_compiler/akg/akg_kernel_json_generator.h

+ 10
- 3
mindspore/ccsrc/backend/kernel_compiler/akg/akg_kernel_json_generator.cc View File

@@ -575,11 +575,18 @@ bool AkgKernelJsonGenerator::CollectFusedJson(const std::vector<AnfNodePtr> &anf
auto fg = anf_nodes[0]->func_graph();
MS_EXCEPTION_IF_NULL(fg);
auto attr_val = fg->get_attr(FUNC_GRAPH_ATTR_GRAPH_KERNEL);
constexpr size_t name_len_limited = 80;
if (attr_val != nullptr) {
auto fg_attr = GetValue<std::string>(attr_val);
(void)kernel_name_.append(fg_attr).append("_");
auto fg_name = GetValue<std::string>(attr_val);
if (fg_name.size() > name_len_limited) {
(*kernel_json)[kJsonKeyOpFullName] = kernel_name_ + fg_name;
auto suffix_pos = fg_name.find_last_of("_");
fg_name =
fg_name.substr(0, name_len_limited - fg_name.size() + suffix_pos) + "_more" + fg_name.substr(suffix_pos);
}
static_cast<void>(kernel_name_.append(fg_name).append("_"));
}
(void)kernel_name_.append(std::to_string(hash_id));
static_cast<void>(kernel_name_.append(std::to_string(hash_id)));
(*kernel_json)[kJsonKeyId] = GetOpCntInc();
(*kernel_json)[kJsonKeyOp] = kernel_name_;
(*kernel_json)[kJsonKeyPlatform] = "AKG";


+ 1
- 0
mindspore/ccsrc/backend/kernel_compiler/akg/akg_kernel_json_generator.h View File

@@ -48,6 +48,7 @@ constexpr auto kJsonKeyOp = "op";
constexpr auto kJsonKeyPtrAddress = "ptr_address";
constexpr auto kJsonKeyCompositeGraph = "composite_graph";
constexpr auto kJsonKeyPlatform = "platform";
constexpr auto kJsonKeyOpFullName = "op_full_name";

constexpr auto kAttrInputNames = "input_names";



Loading…
Cancel
Save