Using TMA in CuteDSL

Some time ago, I explored using CuteDSL to write some kernels for vLLM to enjoy the JIT benefits (ability to specialize on more parameters, don&rsquo;t need to recompile vLLM from source during development, etc&hellip;). I have always known that we can write CUDA C++ way (SIMT style) in CuteDSL: there are cute.arch.thread_idx() / cute.arch.block_idx() (equivalent of threadIdx / blockIdx in CUDA…