xla
[Pallas] Add a bfloat16 flash attention test case
#6810

Merged

[Pallas] Add a bfloat16 flash attention test case #6810

lsy323 merged 1 commit into r2.3 from alanwaketan/bp_1

Initial commit

92d1f98a

alanwaketan requested a review from

will-cromar 1 year ago

alanwaketan requested a review from

JackCaoG 1 year ago

alanwaketan requested a review from

yeounoh 1 year ago

alanwaketan requested a review from

mateuszlewko 1 year ago

alanwaketan requested a review from

stgpetrovic 1 year ago

alanwaketan changed the base branch from master to r2.3 1 year ago

alanwaketan requested a review from

lsy323 1 year ago

JackCaoG approved these changes on 2024-03-22

lsy323 approved these changes on 2024-03-25

lsy323 merged 3522be16 into r2.3 1 year ago

Reviewers

lsy323

JackCaoG

will-cromar

yeounoh

mateuszlewko

stgpetrovic

Assignees

No one assigned

Labels

None yet

Milestone

No milestone