Shengyu Liu interestingLSY
interestingLSY
/ element_stride.cu
Created
August 15, 2025 08:08
Meaning of "elementStride" in TMA descriptor
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #include <cstdio> | |
| #include <cuda_runtime.h> | |
| #include <cuda.h> | |
| #include <cute/arch/copy_sm90_tma.hpp> | |
| #include "utils.h" | |
| __global__ void test_kernel(const __grid_constant__ CUtensorMap tensor_map) { | |
| __shared__ int data[16][16]; |