Commit 4a5b2257 authored by Rosty Geyyer's avatar Rosty Geyyer
Browse files

Format

parent 3e002b60
...@@ -789,7 +789,7 @@ struct DeviceBatchedGemmMultipleDGemmMultipleD_Xdl_CShuffle ...@@ -789,7 +789,7 @@ struct DeviceBatchedGemmMultipleDGemmMultipleD_Xdl_CShuffle
} }
// check if DsLayout is supported // 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 CheckDLayout()
{ {
static bool valid = true; static bool valid = true;
...@@ -820,8 +820,10 @@ struct DeviceBatchedGemmMultipleDGemmMultipleD_Xdl_CShuffle ...@@ -820,8 +820,10 @@ struct DeviceBatchedGemmMultipleDGemmMultipleD_Xdl_CShuffle
is_same_v<tensor_layout::gemm::ColumnMajor, B0Layout> && is_same_v<tensor_layout::gemm::ColumnMajor, B0Layout> &&
CheckDLayout<tensor_layout::gemm::RowMajor, D0sLayout, NumD0Tensor>() && CheckDLayout<tensor_layout::gemm::RowMajor, D0sLayout, NumD0Tensor>() &&
(is_same_v<tensor_layout::gemm::RowMajor, B1Layout> || (is_same_v<tensor_layout::gemm::RowMajor, B1Layout> ||
is_same_v<tensor_layout::gemm::ColumnMajor, B1Layout>) && is_same_v<tensor_layout::gemm::ColumnMajor,
CheckDLayout<tensor_layout::gemm::RowMajor, D1sLayout, NumD1Tensor>() && B1Layout>)&&CheckDLayout<tensor_layout::gemm::RowMajor,
D1sLayout,
NumD1Tensor>() &&
is_same_v<tensor_layout::gemm::RowMajor, E1Layout>)) is_same_v<tensor_layout::gemm::RowMajor, E1Layout>))
{ {
return false; return false;
......
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