Commit 34c4c24a authored by Paul's avatar Paul
Browse files

Format

parent cfec69d3
......@@ -25,7 +25,7 @@ struct ck_gemm
void check_gemm_shape(const shape& s) const
{
if(not contains(range(s.strides().rbegin(), s.strides().rbegin()+3), 1))
if(not contains(range(s.strides().rbegin(), s.strides().rbegin() + 3), 1))
MIGRAPHX_THROW("Invalid shape for ck_gemm");
}
......@@ -80,7 +80,7 @@ struct find_ck_gemm
auto gemm_it = std::find(inputs.begin(), inputs.end(), x_ins);
auto gemm_idx = gemm_it - inputs.begin();
assert(gemm_it != inputs.end());
if (ins->get_shape().type() != shape::half_type)
if(ins->get_shape().type() != shape::half_type)
return;
if(gemm_idx != 0)
{
......
......@@ -233,9 +233,10 @@ struct ck_gemm_compiler : compiler<ck_gemm_compiler>
ip.set_gemm("ck::tensor_operation::device::GemmSpecialization::" + gemm_type);
auto blocks_per_batch = ip.get_grid_size(config);
auto batch_count = std::accumulate(
c_shape.lens().rbegin() + 2, c_shape.lens().rend(), std::size_t{1}, std::multiplies<std::size_t>());
auto batch_count = std::accumulate(c_shape.lens().rbegin() + 2,
c_shape.lens().rend(),
std::size_t{1},
std::multiplies<std::size_t>());
hip_compile_options options;
auto block_size = ip.get_block_size();
......
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