Skip to content
Build and test

[Pallas] Support segment ids in flash attention #8021

Sign in to view logs

[Pallas] Support segment ids in flash attention

[Pallas] Support segment ids in flash attention #8021

CPU tests  /  test (python_tests, xla_op2)

CPU tests / test (python_tests, xla_op2)
succeeded Apr 30, 2024 in 2m 37s