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

bug fix (#47611)

上级 e67d6f17
......@@ -155,12 +155,14 @@ int PrelnResidualBiasFusePass::ApplyPattern(ir::Graph *graph,
x = gpd.mutable_pattern()
->NewNode("preln_residual_bias_fuse/x")
->AsInput()
->assert_is_op_input("elementwise_add", "X");
->assert_is_op_input("elementwise_add", "X")
->assert_var_not_persistable();
y = gpd.mutable_pattern()
->NewNode("preln_residual_bias_fuse/y")
->AsInput()
->assert_is_op_input("elementwise_add", "Y");
->assert_is_op_input("elementwise_add", "Y")
->assert_var_not_persistable();
}
patterns::PrelnResidualBias fused_pattern(
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.
先完成此消息的编辑!
想要评论请 注册