Run unit tests on GPUs

Fixing CI for pytorch version tests (#3130) #441

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
gpu-tests (pytorch)	27m 28s
gpu-tests (pytorch-nightly)	31m 17s
	58m 45s