PR GPU tests

Fix pairwise attention comparison in test #3126

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
gpu-2.0.1 / pytest-gpu	14m 40s
gpu-latest / pytest-gpu	15m 0s
gpu-2.1.0 / pytest-gpu	14m 58s
gpu-2.1.0-flash2 / pytest-gpu	14m 58s
	59m 36s