Skip to content

GitLab

Commit 05e89700 authored by Paul's avatar Paul
Browse files

Format

parent b41b7699
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
src/targets/gpu/fuse_ck.cpp
View file @ 05e89700
...@@ -111,9 +111,9 @@ struct find_ck_gemm_pointwise ...@@ -111,9 +111,9 @@ struct find_ck_gemm_pointwise
auto gemm_idx = gemm_it - inputs.begin(); auto gemm_idx = gemm_it - inputs.begin();
if(ins->get_shape().type() != gemm_ins->get_shape().type()) if(ins->get_shape().type() != gemm_ins->get_shape().type())
return; return;
if (std::any_of(ins->inputs().begin(), ins->inputs().end(), [](auto input) { if(std::any_of(ins->inputs().begin(), ins->inputs().end(), [](auto input) {
return not is_ck_supported_type(input->get_shape().type()); return not is_ck_supported_type(input->get_shape().type());
})) }))
return; return;
assert(gemm_it != inputs.end()); assert(gemm_it != inputs.end());
if(gemm_idx != 0) if(gemm_idx != 0)
......
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