Skip to content

GitLab

Commit a4e2ee8f authored by Paul's avatar Paul
Browse files

Format

parent 8d02a27a
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/targets/gpu/fuse_ck.cpp
View file @ a4e2ee8f
...@@ -54,8 +54,8 @@ MIGRAPHX_PRED_MATCHER(is_ck_gemm, instruction_ref ins) ...@@ -54,8 +54,8 @@ MIGRAPHX_PRED_MATCHER(is_ck_gemm, instruction_ref ins)
if(ins->name() != "dot" and ins->name() != "quant_dot") if(ins->name() != "dot" and ins->name() != "quant_dot")
return false; return false;
if(not contains({shape::half_type, shape::int8_type, shape::int32_type}, if(not contains({shape::half_type, shape::int8_type, shape::int32_type},
ins->get_shape().type())) ins->get_shape().type()))
return false; return false;
auto a = ins->inputs().front()->get_shape(); auto a = ins->inputs().front()->get_shape();
auto b = ins->inputs().back()->get_shape(); auto b = ins->inputs().back()->get_shape();
if(a.lens().back() > 2048) if(a.lens().back() > 2048)
......
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