未验证 提交 5160628c 编写于 作者: W wenbin 提交者: GitHub

bug fix (#47611)

上级 e67d6f17
...@@ -155,12 +155,14 @@ int PrelnResidualBiasFusePass::ApplyPattern(ir::Graph *graph, ...@@ -155,12 +155,14 @@ int PrelnResidualBiasFusePass::ApplyPattern(ir::Graph *graph,
x = gpd.mutable_pattern() x = gpd.mutable_pattern()
->NewNode("preln_residual_bias_fuse/x") ->NewNode("preln_residual_bias_fuse/x")
->AsInput() ->AsInput()
->assert_is_op_input("elementwise_add", "X"); ->assert_is_op_input("elementwise_add", "X")
->assert_var_not_persistable();
y = gpd.mutable_pattern() y = gpd.mutable_pattern()
->NewNode("preln_residual_bias_fuse/y") ->NewNode("preln_residual_bias_fuse/y")
->AsInput() ->AsInput()
->assert_is_op_input("elementwise_add", "Y"); ->assert_is_op_input("elementwise_add", "Y")
->assert_var_not_persistable();
} }
patterns::PrelnResidualBias fused_pattern( patterns::PrelnResidualBias fused_pattern(
gpd.mutable_pattern(), "preln_residual_bias_fuse", with_bias); gpd.mutable_pattern(), "preln_residual_bias_fuse", with_bias);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册