mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

fixed 120e8e3 from https://gitee.com/reku1997/mindspore/pulls/45969 

speed up reshapre transpose fusion pass
This commit is contained in:
reku1997 2022-11-24 15:43:33 +08:00 committed by 王禹程
parent 74662a9083
commit 4d25617f96
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/reshape_transpose_fusion.cc
View File

@ -48,9 +48,8 @@ bool CheckMatmulNeighborNodes(const FuncGraphPtr &func_graph, const AnfNodePtr &
auto manager = func_graph->manager();
MS_EXCEPTION_IF_NULL(manager);
auto user_list = manager->node_users();
auto iter = user_list.find(down_node);
if (iter != user_list.end()) {
auto iter = manager->node_users().find(down_node);
if (iter != manager->node_users().end()) {
if (std::any_of(iter->second.begin(), iter->second.end(), [](const std::pair<AnfNodePtr, int64_t> &kernel) {
return IsPrimitiveCNode(kernel.first, prim::kPrimMatMul) ||
IsPrimitiveCNode(kernel.first, prim::kPrimBatchMatMul);