|
|
|
@@ -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"; |
|
|
|
|