diff --git a/ge/ir_build/attr_options/weight_compress_option.cc b/ge/ir_build/attr_options/weight_compress_option.cc index 06528af2..0b8af37e 100644 --- a/ge/ir_build/attr_options/weight_compress_option.cc +++ b/ge/ir_build/attr_options/weight_compress_option.cc @@ -46,18 +46,23 @@ graphStatus WeightCompressFunc(ComputeGraphPtr &graph, const string &cfg_path) { vector compress_node_vec = StringUtils::Split(compress_nodes, ';'); for (size_t i = 0; i < compress_node_vec.size(); ++i) { + bool is_find = false; for (auto &node_ptr : graph->GetDirectNode()) { GE_CHECK_NOTNULL(node_ptr); auto op_desc = node_ptr->GetOpDesc(); GE_CHECK_NOTNULL(op_desc); if ((op_desc->GetName() == compress_node_vec[i]) || IsOriginalOpFind(op_desc, compress_node_vec[i])) { + is_find = true; if (!ge::AttrUtils::SetBool(op_desc, ge::ATTR_NAME_COMPRESS_WEIGHT, true)) { GELOGE(GRAPH_FAILED, "node %s SetBool failed.", compress_node_vec[i].c_str()); return GRAPH_FAILED; } } } + if (!is_find) { + GELOGW("node %s is not in graph", compress_node_vec[i].c_str()); + } } return GRAPH_SUCCESS; } diff --git a/metadef b/metadef index 573c7983..29a779d0 160000 --- a/metadef +++ b/metadef @@ -1 +1 @@ -Subproject commit 573c7983273efa902e246052382090d3bc3e5804 +Subproject commit 29a779d0b1e0482092e387ccce2886b36dc5e395