Fix moving keys to device in ResultCollection #13704

Job	Run time
pl-cpu (macOS-14, lightning, 3.10, 2.1)	14m 51s
pl-cpu (ubuntu-20.04, lightning, 3.10, 2.1)	14m 40s
pl-cpu (windows-2022, lightning, 3.10, 2.1)	16m 47s
pl-cpu (macOS-14, lightning, 3.11, 2.2)	18m 44s
pl-cpu (ubuntu-20.04, lightning, 3.11, 2.2)	13m 59s
pl-cpu (windows-2022, lightning, 3.11, 2.2)	18m 19s
pl-cpu (macOS-14, lightning, 3.11, 2.3)	19m 9s
pl-cpu (ubuntu-20.04, lightning, 3.11, 2.3)	14m 16s
pl-cpu (windows-2022, lightning, 3.11, 2.3)	17m 56s
pl-cpu (macOS-14, lightning, 3.12, 2.4)	16m 11s
pl-cpu (ubuntu-20.04, lightning, 3.12, 2.4)	20m 38s
pl-cpu (windows-2022, lightning, 3.12, 2.4)	17m 15s
pl-cpu (macOS-12, pytorch, 3.10, 2.1)	27m 42s
pl-cpu (ubuntu-22.04, pytorch, 3.10, 2.1)	14m 35s
pl-cpu (windows-2022, pytorch, 3.10, 2.1)	17m 19s
pl-cpu (macOS-13, lightning, 3.9, 2.1, oldest)	25m 4s
pl-cpu (ubuntu-20.04, lightning, 3.9, 2.1, oldest)	12m 24s
pl-cpu (windows-2022, lightning, 3.9, 2.1, oldest)	13m 59s
pl-cpu (macOS-14, pytorch, 3.9, 2.1)	16m 32s
pl-cpu (ubuntu-20.04, pytorch, 3.9, 2.1)	14m 55s
pl-cpu (windows-2022, pytorch, 3.9, 2.1)	17m 55s
	6h 3m 10s

Provide feedback