xla
db7112af - [Backport] Make a FlashAttention Wrapper (#6827)

Commit

2 years ago

[Backport] Make a FlashAttention Wrapper (#6827)

References

#6827 - [Backport] Make a FlashAttention Wrapper

Author

alanwaketan

alanwaketan

Parents

Loading