Skip to content

[Backport] Make a FlashAttention Wrapper #6618

[Backport] Make a FlashAttention Wrapper

[Backport] Make a FlashAttention Wrapper #6618

Annotations

1 warning

GPU tests  /  test (python_tests, xla_op2)

succeeded Mar 27, 2024 in 46m 57s