Use thread 0 for all computation (slow but correct). SMEM for Q and O sharing across threads. Online softmax with O rescale — correct D1.5 approach. D3 SWA mask implemented. Target: cos ~0.999998 then parallelize.
Use thread 0 for all computation (slow but correct). SMEM for Q and O sharing across threads. Online softmax with O rescale — correct D1.5 approach. D3 SWA mask implemented. Target: cos ~0.999998 then parallelize.