Browse Source

!579 Bugfix: fix fusion error in transop breadth fusion pass

From: @lichun30
Reviewed-by: @sheng-nan,@ji_chen
Signed-off-by: @ji_chen
tags/v1.2.0
mindspore-ci-bot Gitee 3 years ago
parent
commit
5aa5b05525
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      ge/graph/passes/transop_breadth_fusion_pass.cc

+ 3
- 1
ge/graph/passes/transop_breadth_fusion_pass.cc View File

@@ -70,8 +70,10 @@ std::string TransOpBreadthFusionPass::GetNodeId(const int anchor_index, const No
trans_data_type = true;
trans_format = true;
trans_shape = true;
} else if (node->GetType() == RESHAPE) {
} else if (node->GetType() == RESHAPE || node->GetType() == EXPANDDIMS || node->GetType() == SQUEEZE) {
trans_shape = true;
} else if (node->GetType() == REFORMAT) {
trans_format = true;
}

id << node->GetType() << '-' << anchor_index;


Loading…
Cancel
Save