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>
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 b_shape = inputs[1];
auto c_shape = inputs.back();
auto rank = a_shape.lens().size();
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