Format

src/targets/gpu/jit/ck_gemm.cpp

@@ -170,14 +170,11 @@ static std::size_t get_tuning_for(const std::vector<shape>& inputs)
 
 struct ck_gemm_compiler : compiler<ck_gemm_compiler>
 {
-    static bool transposed_matrix(const shape& s)
-    {
-        return s.strides().back() != 1;
-    }
+    static bool transposed_matrix(const shape& s) { return s.strides().back() != 1; }
     static std::string get_layout(const shape& s)
     {
         return transposed_matrix(s) ? "ck::tensor_layout::gemm::ColumnMajor"
-                              : "ck::tensor_layout::gemm::RowMajor";
+                                    : "ck::tensor_layout::gemm::RowMajor";
     }
 
     static std::string get_type(const shape& s)
@@ -197,9 +194,9 @@ struct ck_gemm_compiler : compiler<ck_gemm_compiler>
 
     static std::vector<shape> adjust_inputs(std::vector<shape> inputs, bool& swap_inputs)
     {
-        swap_inputs = false;
+        swap_inputs  = false;
         auto c_shape = inputs.back();
-        if (not transposed_matrix(c_shape))
+        if(not transposed_matrix(c_shape))
             return inputs;
         std::vector<int64_t> perm(c_shape.lens().size());
         std::iota(perm.begin(), perm.end(), 0);