Commit e665173

authored and

The jax_triton Authors

committed

Add segment_ids support to pallas flash attention on GPU.

PiperOrigin-RevId: 547592877

1 parent 321a19e commit e665173Copy full SHA for e665173

1 file changed

+2019

-0

lines changed

+2019

-0

lines changed

Comments

(0)