[PyTorch] Miscellaneous fixes for FA3 attention #5472

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
Core	25m 34s
PaddlePaddle	23m 27s
JAX	25m 9s
PyTorch	1h 14m 57s
	2h 29m 7s