Creează un cont pentru a te abona la acest blog și a-i urmări articolele pe pagina Lectură.
Articole: 4
-
Sequel to "The kernel that wasn't there".
-
Or: what 7% taught me before I wrote a single kernel.
-
I wrote a fused decode-attention kernel for an RL training loop, got it 2.2× faster than the SDPA path it replaces at the microbenchmark level, dropped it into HuggingFace's generate, and watched the decode step get nearly 3× slower.
-
After writing Flash Attention 1 in plain CUDA, I wanted to take the next step and port Flash Attention 2 to CuteDSL, the Python front-end NVIDIA ships with CUTLASS 4.