"...composable_kernel_onnxruntime.git" did not exist on "fd49ff8080b90687108c46f92321ce10ecc743dc"
Commit 515a9096 authored by turneram's avatar turneram
Browse files

Use div in matcher instead of mul

parent 9b7a906d
...@@ -853,13 +853,19 @@ struct find_div_const ...@@ -853,13 +853,19 @@ struct find_div_const
struct find_gelu_erf struct find_gelu_erf
{ {
static auto match_div()
{
return match::name("div")(
args(match::any().bind("x"), match::skip_broadcasts(match::is_constant())));
}
static auto match_mul1() static auto match_mul1()
{ {
return match::name("mul")( return match::name("mul")(
args(match::any().bind("x"), match::skip_broadcasts(match::name("recip")))); args(match::any().bind("x"), match::skip_broadcasts(match::name("recip"))));
} }
static auto match_erf() { return match::name("erf")(match::arg(0)(match_mul1())); } static auto match_erf() { return match::name("erf")(match::arg(0)(match_div())); }
static auto match_add2() static auto match_add2()
{ {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment