xla
[Backport] Allow setting FlashAttention's causal mask
#6837

Merged

[Backport] Allow setting FlashAttention's causal mask #6837

alanwaketan merged 1 commit into r2.3 from alanwaketan/backport-1

Initial commit

ac3d4059

alanwaketan requested a review from

lsy323 1 year ago

alanwaketan requested a review from

JackCaoG 1 year ago

JackCaoG approved these changes on 2024-03-27

lsy323 approved these changes on 2024-03-27

alanwaketan merged 046f9103 into r2.3 1 year ago

alanwaketan deleted the alanwaketan/backport-1 branch 1 year ago

Reviewers

lsy323

JackCaoG

Assignees

No one assigned

Labels

None yet

Milestone

No milestone