p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix fusion condition of transpose and reshape

This commit is contained in:
yujianfeng 2020-05-26 16:04:35 +08:00
parent 7f80d02807
commit 94818cf255
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/ccsrc/pre_activate/ascend/ir_fusion/transpose_reshape_fusion.cc
View File

@ -50,9 +50,9 @@ const AnfNodePtr TransposeReshapeFusion::Process(const FuncGraphPtr &func_graph,
MS_EXCEPTION_IF_NULL(reshape_cnode); MS_EXCEPTION_IF_NULL(reshape_cnode);
auto transpose_cnode = CheckAnfNodeIfCNodeAndInputSize(reshape_cnode->input(1), kBackendReshapeInputNum); auto transpose_cnode = CheckAnfNodeIfCNodeAndInputSize(reshape_cnode->input(1), kBackendReshapeInputNum);
MS_EXCEPTION_IF_NULL(transpose_cnode); MS_EXCEPTION_IF_NULL(transpose_cnode);
std::vector<size_t> reshape_input0_shape = AnfAlgo::GetPrevNodeOutputInferShape(reshape_cnode, 0); std::vector<size_t> reshape_output0_shape = AnfAlgo::GetOutputInferShape(reshape_cnode, 0);
std::vector<size_t> transpose_input0_shape = AnfAlgo::GetPrevNodeOutputInferShape(transpose_cnode, 0); std::vector<size_t> transpose_input0_shape = AnfAlgo::GetPrevNodeOutputInferShape(transpose_cnode, 0);
if (!CheckShapeDimInfo(reshape_input0_shape) || !CheckShapeDimInfo(transpose_input0_shape)) { if (!CheckShapeDimInfo(reshape_output0_shape) || !CheckShapeDimInfo(transpose_input0_shape)) {
return nullptr; return nullptr;
} }
auto prim = std::make_shared<Primitive>(kConfusionTransposeDOpName); auto prim = std::make_shared<Primitive>(kConfusionTransposeDOpName);

2
tests/ut/cpp/python_input/gtest_input/pre_activate/transpose_reshape_fusion_test.py
View File

@ -38,7 +38,7 @@ def test_transpose_reshape_fusion(tag):
@fns @fns
def before(x): def before(x):
transpose = Transpose(x, (1, 0, 2, 3)) transpose = Transpose(x, (1, 0, 2, 3))
reshape = Reshape(transpose, (2, 4, 8, 16)) reshape = Reshape(transpose, (2, 2, 16, 16))
return reshape return reshape
@fns @fns