leavelet/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 24 Packages Projects Releases Wiki Activity

Facebook:This commit makes its files safe for use with -Wimplicit-fallthrough. (#2324)

Browse Source
This commit is contained in:
Wenxin Cheng
2025-07-31 20:55:19 -04:00
committed by GitHub
parent 3b054767b3
commit 6dd13d4278
2 changed files with 9 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/cute/atom/copy_traits_sm90_tma_swizzle.hpp
+5 -7
View File
@@ -48,13 +48,11 @@ TMA::SmemSwizzleBits
get_tma_swizzle_bits(Swizzle<B,M,S>)
{
if constexpr (M == 4) {
switch (B) {
default: static_assert(0 <= B && B <= 3, "Expected B = 0,1,2, or 3 when M == 4. Unsupported layout swizzle.");
case 3: return TMA::SmemSwizzleBits::B128;
case 2: return TMA::SmemSwizzleBits::B64;
case 1: return TMA::SmemSwizzleBits::B32;
case 0: return TMA::SmemSwizzleBits::DISABLE;
}
static_assert(0 <= B && B <= 3, "Expected B = 0,1,2, or 3 when M == 4. Unsupported layout swizzle.");
if constexpr (B == 3) { return TMA::SmemSwizzleBits::B128; }
if constexpr (B == 2) { return TMA::SmemSwizzleBits::B64; }
if constexpr (B == 1) { return TMA::SmemSwizzleBits::B32; }
if constexpr (B == 0) { return TMA::SmemSwizzleBits::DISABLE; }
} else
if constexpr (M == 5 || M == 6) {