4 Commits

Author	SHA1	Message	Date
Angelos Katharopoulos	f5cc1eea72	Allow different value dimensions in sdpa_vector (#1811)	2025-01-31 20:58:59 -08:00
Awni Hannun	d1766f2c70	Add boolean mask support in vector SDPA (#1757)	2025-01-07 20:24:53 -08:00
Angelos Katharopoulos	073076ac7d	2-Pass Sdpa Inference Kernel (#1597)	2024-11-18 17:31:53 -08:00
Angelos Katharopoulos	50d8bed468	Fused attention for single query (#1497)	2024-10-18 00:58:52 -07:00