Commit 79eac1b8 authored by Paul's avatar Paul
Browse files

Format

parent 4d50c338
......@@ -270,8 +270,10 @@ struct find_dot_mul
auto matcher() const
{
auto const_broadcast = match::name("broadcast", "multibroadcast")(match::is_constant());
auto mul = match::name("mul")(match::used_once(), match::either_arg(0, 1)(
const_broadcast.bind("d"), match::none_of(match::is_constant()).bind("z")));
auto mul = match::name("mul")(
match::used_once(),
match::either_arg(0, 1)(const_broadcast.bind("d"),
match::none_of(match::is_constant()).bind("z")));
return match::name("dot")(match::either_arg(0, 1)(mul, match::is_constant().bind("c")));
}
......
......@@ -770,7 +770,8 @@ struct find_contiguous_pointwise
{
auto matcher() const
{
return match::name("gpu::contiguous")(match::arg(0)(precompile_name("pointwise")(match::used_once())));
return match::name("gpu::contiguous")(
match::arg(0)(precompile_name("pointwise")(match::used_once())));
}
void apply(module& m, const match::matcher_result& r) const
......@@ -790,7 +791,8 @@ struct find_layernorm_pointwise
auto matcher() const
{
return precompile_name("pointwise")(match::arg(0)(
precompile_name("gpu::prelayernorm", "gpu::preadd_layernorm")(match::used_once()).bind("layernorm")));
precompile_name("gpu::prelayernorm", "gpu::preadd_layernorm")(match::used_once())
.bind("layernorm")));
}
void apply(module& m, const match::matcher_result& r) const
......
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