Format

include/ck/tensor_operation/gpu/device/impl/device_batched_gemm_multiple_d_gemm_multiple_d_xdl_cshuffle.hpp

@@ -789,13 +789,13 @@ struct DeviceBatchedGemmMultipleDGemmMultipleD_Xdl_CShuffle
     }
 
     // check if DsLayout is supported
-    template<typename RefLayout, typename DsLayout, const index_t NumDTensor>
+    template <typename RefLayout, typename DsLayout, const index_t NumDTensor>
     static bool CheckDLayout()
     {
         static bool valid = true;
         // iterate over DLayout tuple
         static_for<0, NumDTensor, 1>{}([&](auto i) {
-            using DLayout   = remove_cvref_t<tuple_element_t<i.value, DsLayout>>;
+            using DLayout = remove_cvref_t<tuple_element_t<i.value, DsLayout>>;
             // if RefLayout and DLayout are same, keep valid true, otherwise false
             valid = valid && is_same_v<RefLayout, DLayout>;
         });
@@ -816,12 +816,14 @@ struct DeviceBatchedGemmMultipleDGemmMultipleD_Xdl_CShuffle
         // B1 - Row or Col
         // D1s - Rows
         // E1 - Row
-        if(!(is_same_v<tensor_layout::gemm::RowMajor, A0Layout> && 
+        if(!(is_same_v<tensor_layout::gemm::RowMajor, A0Layout> &&
              is_same_v<tensor_layout::gemm::ColumnMajor, B0Layout> &&
              CheckDLayout<tensor_layout::gemm::RowMajor, D0sLayout, NumD0Tensor>() &&
-             (is_same_v<tensor_layout::gemm::RowMajor, B1Layout> || 
-              is_same_v<tensor_layout::gemm::ColumnMajor, B1Layout>) &&
-              CheckDLayout<tensor_layout::gemm::RowMajor, D1sLayout, NumD1Tensor>() &&
+             (is_same_v<tensor_layout::gemm::RowMajor, B1Layout> ||
+              is_same_v<tensor_layout::gemm::ColumnMajor,
+                        B1Layout>)&&CheckDLayout<tensor_layout::gemm::RowMajor,
+                                                 D1sLayout,
+                                                 NumD1Tensor>() &&
              is_same_v<tensor_layout::gemm::RowMajor, E1Layout>))
         {
             return false;