From 182efca094b3299e23e663d55042697493ddf9dc Mon Sep 17 00:00:00 2001 From: jjfeing Date: Thu, 17 Dec 2020 17:20:48 +0800 Subject: [PATCH] fix fusion build bug --- mindspore/ccsrc/backend/kernel_compiler/kernel_fusion.cc | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/mindspore/ccsrc/backend/kernel_compiler/kernel_fusion.cc b/mindspore/ccsrc/backend/kernel_compiler/kernel_fusion.cc index 702d232b45..c6026479e8 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/kernel_fusion.cc +++ b/mindspore/ccsrc/backend/kernel_compiler/kernel_fusion.cc @@ -41,8 +41,10 @@ static size_t GenFusionJsonHash(const nlohmann::json &fusion_json) { for (auto &input_desc : op["input_desc"]) { input_desc.erase("name"); } - for (auto &list_arg : op["prebuild_output_attrs"]["list_args"]) { - list_arg.erase("name"); + for (auto &list_arg : op["prebuild_outs_attrs"]["list_args"]) { + if (list_arg.is_object() && list_arg.find("name") != list_arg.end()) { + list_arg.erase("name"); + } } } }