PR GPU tests

Attempt to fix flaky test #2713

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
gpu-2.0.1 / pytest-gpu	5m 51s
gpu-2.1.0 / pytest-gpu	6m 7s
gpu-2.1.0-flash2 / pytest-gpu	6m 6s
gpu-latest / pytest-gpu	6m 4s
	24m 8s