Browse Source

fix batchmatmul fusion and sub graph bugs

tags/v1.1.0
zengxianglong 5 years ago
parent
commit
2b91ef7fd5
2 changed files with 7 additions and 0 deletions
  1. +4
    -0
      mindspore/lite/src/sub_graph_kernel.cc
  2. +3
    -0
      mindspore/lite/tools/optimizer/fusion/batchmatmul_fusion.cc

+ 4
- 0
mindspore/lite/src/sub_graph_kernel.cc View File

@@ -147,6 +147,10 @@ int SubGraphKernel::ReSize(bool is_interrupt) {
}
}
}
if (is_interrupt) {
MS_LOG(INFO) << "Infer shape failed.";
return RET_INFER_INVALID;
}
return RET_OK;
}



+ 3
- 0
mindspore/lite/tools/optimizer/fusion/batchmatmul_fusion.cc View File

@@ -133,6 +133,9 @@ const AnfNodePtr BatchMatMulFusion::Process(const FuncGraphPtr &func_graph, cons
MS_ASSERT(fullconnect_cnode->inputs().size() == 3);
auto left_slice_node = fullconnect_cnode->input(1);
auto left_slice_cnode = left_slice_node->cast<CNodePtr>();
if (GetCNodeType(left_slice_cnode) != schema::PrimitiveType_Slice) {
return nullptr;
}
auto left_matmul_input = left_slice_cnode->input(1);
auto right_reshape_node = fullconnect_cnode->input(2);



Loading…
Cancel
Save