Commit 4acc55c3 authored by Alan Turner's avatar Alan Turner
Browse files

Formatting

parent a7d57049
...@@ -173,15 +173,12 @@ struct find_ck_gemm ...@@ -173,15 +173,12 @@ struct find_ck_gemm
struct find_ck_gemm_softmax_gemm struct find_ck_gemm_softmax_gemm
{ {
auto matcher() const auto matcher() const { return match::name("gpu::pre_gemm_softmax_gemm"); }
{
return match::name("gpu::pre_gemm_softmax_gemm");
}
void apply(module_pass_manager& mpm, const match::matcher_result& r) const void apply(module_pass_manager& mpm, const match::matcher_result& r) const
{ {
auto ins = r.result; auto ins = r.result;
auto v = ins->get_operator().to_value(); auto v = ins->get_operator().to_value();
assert(v.contains("scale")); assert(v.contains("scale"));
auto scale = v.at("scale").to<float>(); auto scale = v.at("scale").to<float>();
mpm.get_module().replace_instruction( mpm.get_module().replace_instruction(
......
...@@ -181,7 +181,7 @@ void prefuse_ops::apply(module_pass_manager& mpm) const ...@@ -181,7 +181,7 @@ void prefuse_ops::apply(module_pass_manager& mpm) const
match::find_matches(mpm.get_module(), find_layernorm{}); match::find_matches(mpm.get_module(), find_layernorm{});
mpm.run_pass(dead_code_elimination{}); mpm.run_pass(dead_code_elimination{});
match::find_matches(mpm.get_module(), find_add_layernorm{}); match::find_matches(mpm.get_module(), find_add_layernorm{});
if (enabled(MIGRAPHX_ENABLE_CK{})) if(enabled(MIGRAPHX_ENABLE_CK{}))
match::find_matches(mpm, find_gemm_softmax_gemm{}); match::find_matches(mpm, find_gemm_softmax_gemm{});
} }
......
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