Commit 9b3e4576 authored by Paul's avatar Paul
Browse files

Format

parent 74f21ca6
...@@ -276,13 +276,13 @@ struct ck_gemm_compiler : compiler<ck_gemm_compiler> ...@@ -276,13 +276,13 @@ struct ck_gemm_compiler : compiler<ck_gemm_compiler>
return rank >= 3 and b_strides[rank - 3] == 0; return rank >= 3 and b_strides[rank - 3] == 0;
} }
ck::host::device_gemm_multiple_d::Problem create_problem(const std::vector<shape>& inputs, const value& v) const ck::host::device_gemm_multiple_d::Problem create_problem(const std::vector<shape>& inputs,
const value& v) const
{ {
auto a_shape = inputs[0]; auto a_shape = inputs[0];
auto b_shape = inputs[1]; auto b_shape = inputs[1];
auto c_shape = inputs.back(); auto c_shape = inputs.back();
auto rank = a_shape.lens().size(); auto rank = a_shape.lens().size();
auto b_strides = b_shape.strides(); auto b_strides = b_shape.strides();
......
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