4 Commits

Author	SHA1	Message	Date
Tri Dao	a1f49a2b92	[Compilation] Change BOOL_SWITCH to fix Windows compilation ... Follow xFormers's DISTPATCH_BOOL. Haven't tested it on Windows.	2023-01-06 14:40:58 -08:00
Tri Dao	8a2ece89f7	Simplify BOOL_SWITCH macro to fix compiling error on gcc 7	2022-12-06 14:38:32 -08:00
Tri Dao	5badfb7848	Implement attention kernel that splits the batch into two	2022-10-13 20:49:02 -07:00
Tri Dao	de19de7ab1	Implement for bf16	2022-07-09 23:31:56 -07:00