xla
[Backport] Make a FlashAttention Wrapper
#6827
Merged

[Backport] Make a FlashAttention Wrapper #6827

lsy323 merged 1 commit into r2.3 from alanwaketan/backport
alanwaketan
alanwaketan tmp
ed5254e3
alanwaketan alanwaketan requested a review from lsy323 lsy323 1 year ago
alanwaketan alanwaketan requested a review from JackCaoG JackCaoG 1 year ago
JackCaoG
JackCaoG approved these changes on 2024-03-27
alanwaketan
alanwaketan
lsy323
lsy323 approved these changes on 2024-03-27
lsy323 lsy323 merged db7112af into r2.3 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone