Run unit tests on GPUs

Skip unrolling follow up (#3260) #550

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
gpu-tests (pytorch)	8m 43s
gpu-tests (pytorch-nightly)	8m 57s
	17m 40s