"vscode:/vscode.git/clone" did not exist on "77c95479dbc4f7affeb0c6a22af5b8a413ee83fa"
Commit 29aae082 authored by Alan Turner's avatar Alan Turner
Browse files

Formatting

parent f19e41bb
...@@ -177,8 +177,8 @@ struct find_ck_gemm_softmax_gemm ...@@ -177,8 +177,8 @@ struct find_ck_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(
......
...@@ -129,10 +129,10 @@ struct ck_gemm_softmax_gemm_compiler : compiler<ck_gemm_softmax_gemm_compiler> ...@@ -129,10 +129,10 @@ struct ck_gemm_softmax_gemm_compiler : compiler<ck_gemm_softmax_gemm_compiler>
operation compile_op(context& ctx, const std::vector<shape>& inputs, const value& v) const operation compile_op(context& ctx, const std::vector<shape>& inputs, const value& v) const
{ {
const auto& c_shape = inputs.back(); const auto& c_shape = inputs.back();
auto tuning_value = v.get("tuning_value", 5); auto tuning_value = v.get("tuning_value", 5);
auto batch_count = get_batch_count(c_shape); auto batch_count = get_batch_count(c_shape);
auto problem = create_problem(inputs, v); auto problem = create_problem(inputs, v);
const auto include_header = problem.GetIncludeHeader(); const auto include_header = problem.GetIncludeHeader();
const auto solutions = problem.GetSolutions(ctx.get_current_device().get_gfx_name()); const auto solutions = problem.GetSolutions(ctx.get_current_device().get_gfx_name());
......
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