[PyTorch] Miscellaneous fixes for FA3 attention #5488

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
Core	25m 56s
JAX	24m 9s
PyTorch	1h 4m 48s
PaddlePaddle	23m 58s
	2h 18m 51s