Skip to content

[Pallas] Allow setting FlashAttention's causal mask#6792

Merged
alanwaketan merged 2 commits intomasterfrom alanwaketan/flash_attention_4Mar 21, 2024

Commits