GitLab

This is faster since we only need to do atomic adds on dq, instead of atomic
adds on both dk and dv.

They don't have to have the same block size, number of threads, etc.

From 4KB per buffer to 2KB per buffer. This saves us 8KB of smem (each Q and dO
have 2 buffers)