[Example] Using Collector's device args #1858

Job	Run time
generate-matrix / generate	4s
generate-matrix / generate	4s
pytorch/rl / wheel-py3_9-cpu	7m 57s
pytorch/rl / wheel-py3_9-cpu	7m 57s
pytorch/rl / wheel-py3_9-cuda11_8	11m 41s
pytorch/rl / wheel-py3_9-cuda11_8	11m 41s
pytorch/rl / wheel-py3_9-cuda12_4	9m 42s
pytorch/rl / wheel-py3_9-cuda12_4	9m 42s
pytorch/rl / wheel-py3_9-cuda12_6	9m 55s
pytorch/rl / wheel-py3_9-cuda12_6	9m 55s
pytorch/rl / upload / wheel-py3_9-cpu	14s
pytorch/rl / upload / wheel-py3_9-cpu	9s
pytorch/rl / upload / wheel-py3_9-cuda11_8	9s
pytorch/rl / upload / wheel-py3_9-cuda11_8	8s
pytorch/rl / upload / wheel-py3_9-cuda12_4	13s
pytorch/rl / upload / wheel-py3_9-cuda12_4	14s
pytorch/rl / upload / wheel-py3_9-cuda12_6	12s
pytorch/rl / upload / wheel-py3_9-cuda12_6	18s
	1h 20m 15s

Provide feedback