Format

src/targets/gpu/jit/ck_gemm.cpp

@@ -270,14 +270,13 @@ struct ck_gemm_compiler : compiler<ck_gemm_compiler>
     bool can_fold_batch(const std::vector<shape>& inputs) const
     {
         const auto& b_shape = inputs[1];
-        if(std::any_of(inputs.begin() + 2, inputs.end()-1, [](auto input) {
-            return input.broadcasted();
-        }))
+        if(std::any_of(inputs.begin() + 2, inputs.end() - 1, [](auto input) {
+               return input.broadcasted();
+           }))
             return false;
         const auto& b_strides = b_shape.strides();
-        return std::all_of(b_strides.begin(), b_strides.end() - 3, [](auto stride) {
-            return stride == 0;
-        });
+        return std::all_of(
+            b_strides.begin(), b_strides.end() - 3, [](auto stride) { return stride == 0; });
     }
 
     ck::host::device_gemm_multiple_d::Problem create_problem(const std::vector<shape>& inputs,