!8184 fix batchmatmul fusion bug

Merge pull request !8184 from XianglongZeng/myms
pull/8184/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 1e7062e405

@ -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;
}

@ -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