From 60b421c2cab7b879b5ab564dbdb7fde46e97a7ef Mon Sep 17 00:00:00 2001 From: tronzhang <6517937+tronzhang@user.noreply.gitee.com> Date: Fri, 30 Oct 2020 12:52:59 +0800 Subject: [PATCH] limit the length of fuse-op's name --- .../akg/akg_kernel_json_generator.cc | 13 ++++++++++--- .../kernel_compiler/akg/akg_kernel_json_generator.h | 1 + 2 files changed, 11 insertions(+), 3 deletions(-) diff --git a/mindspore/ccsrc/backend/kernel_compiler/akg/akg_kernel_json_generator.cc b/mindspore/ccsrc/backend/kernel_compiler/akg/akg_kernel_json_generator.cc index 0943d7fcfa..15296e2927 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/akg/akg_kernel_json_generator.cc +++ b/mindspore/ccsrc/backend/kernel_compiler/akg/akg_kernel_json_generator.cc @@ -575,11 +575,18 @@ bool AkgKernelJsonGenerator::CollectFusedJson(const std::vector &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(attr_val); - (void)kernel_name_.append(fg_attr).append("_"); + auto fg_name = GetValue(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(kernel_name_.append(fg_name).append("_")); } - (void)kernel_name_.append(std::to_string(hash_id)); + static_cast(kernel_name_.append(std::to_string(hash_id))); (*kernel_json)[kJsonKeyId] = GetOpCntInc(); (*kernel_json)[kJsonKeyOp] = kernel_name_; (*kernel_json)[kJsonKeyPlatform] = "AKG"; diff --git a/mindspore/ccsrc/backend/kernel_compiler/akg/akg_kernel_json_generator.h b/mindspore/ccsrc/backend/kernel_compiler/akg/akg_kernel_json_generator.h index 0b8c01b864..dde8cdbd32 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/akg/akg_kernel_json_generator.h +++ b/mindspore/ccsrc/backend/kernel_compiler/akg/akg_kernel_json_generator.h @@ -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";