3 Commits

Author	SHA1	Message	Date
Ying Zhang	db80387343	Add seqused_q in fwd / bwd and seqused_k in bwd.	2024-09-16 14:24:11 -07:00
Cameron Shinn	3cea2fb6ee	Add ArchTag to pre/postprocess bwd kernels (#1180) ... * Add ArchTag to pre/postprocess bwd kernels * Type-dependent CC check for bwd pre/postprocess * Fix CC >= 90 for bwd postprocess --------- Co-authored-by: Cameron Shinn <cshinn@nvidia.com>	2024-08-28 00:20:47 -07:00
Tri Dao	bafe253042	[FA3] Bwd	2024-08-01 01:57:06 -07:00