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

Commit

1 year ago

[Backport] Make a FlashAttention Wrapper (#6827)

References

#6827 - [Backport] Make a FlashAttention Wrapper

Author

alanwaketan

alanwaketan

Parents

Loading