This is faster since we only need to do atomic adds on dq, instead of atomic adds on both dk and dv.
Attach a file by drag & drop or click to upload
