PR GPU tests

Adding support for alibi when using flash attention #3785

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
gpu-2.1.0 / pytest-gpu	11m 51s
gpu-2.1.0-flash2 / pytest-gpu	10m 35s
	22m 26s