unit test

do `record_data_in_stream` step in `copy_data_to_device` #2508

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
unit_tests_stable_pytorch (3.8)	5m 17s
unit_tests_nightly_pytorch (3.8)	5m 44s
unit_tests_nightly_pytorch (3.9)	5m 52s
	16m 53s