Skip to content
CI

[Pallas] Use scratch_shapes for scratch operands in flash attention kernel. #47499

Sign in to view logs

[Pallas] Use scratch_shapes for scratch operands in flash attention kernel.

[Pallas] Use scratch_shapes for scratch operands in flash attention kernel. #47499

build with 3.9 (py 3.9 on ubuntu-20.04-16core, x64=1)

build with 3.9 (py 3.9 on ubuntu-20.04-16core, x64=1)
succeeded Mar 2, 2024 in 4m 38s