[PyTorch] Miscellaneous fixes for FA3 attention #5471

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
JAX	23m 37s
Core	24m 38s
PaddlePaddle	24m 4s
PyTorch	1h 5m 14s
	2h 17m 33s