xla
3522be16 - [Pallas] Add a bfloat16 flash attention test case (#6810)

Commit

1 year ago

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

References

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

Author

alanwaketan

alanwaketan

Parents

Loading