| 140 | } |
| 141 | |
| 142 | inline CUtensorMapSwizzle |
| 143 | to_CUtensorMapSwizzle(SmemSwizzleBits const& t) { |
| 144 | switch (t) { |
| 145 | default: assert(false && "Unknown SmemSwizzleBits!"); |
| 146 | case SmemSwizzleBits::DISABLE: return CU_TENSOR_MAP_SWIZZLE_NONE; |
| 147 | case SmemSwizzleBits::B32: return CU_TENSOR_MAP_SWIZZLE_32B; |
| 148 | case SmemSwizzleBits::B64: return CU_TENSOR_MAP_SWIZZLE_64B; |
| 149 | case SmemSwizzleBits::B128: return CU_TENSOR_MAP_SWIZZLE_128B; |
| 150 | } |
| 151 | } |
| 152 | |
| 153 | template <int BlockMajorSize, int BlockMinorSize, int TmaDim, typename DType, int B, int M, int S> |
| 154 | TmaDescriptor make_tma_copy_desc( |
no outgoing calls
no test coverage detected