Commit a9b32b71 authored by Alan Turner's avatar Alan Turner
Browse files

Formatting

parent 6d57f78d
......@@ -225,8 +225,9 @@ struct find_ck_gemm_softmax_gemm
{
auto gemm1 =
match::skip(match::name("contiguous"))(match::name("dot")(is_ck_gemm().bind("gemm1")));
auto mul = match::name("pointwise")(match::nargs(2), match::either_arg(0, 1)(
match::is_constant().bind("scale"), gemm1))(is_pointwise_scale());
auto mul = match::name("pointwise")(
match::nargs(2), match::either_arg(0, 1)(match::is_constant().bind("scale"), gemm1))(
is_pointwise_scale());
auto softmax = match::name("softmax")(match::arg(0)(mul)).bind("softmax");
return match::name("dot")(is_ck_gemm().bind("gemm2"))(match::arg(0)(softmax));
......
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