[BugFix] `LazyTensorStorage` only allocates data on the given device

:link: Helpful Links

:test_tube: See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/rl/2188

:page_facing_up: Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

:heavy_exclamation_mark: 1 Active SEVs

There are 1 currently active SEVs. If your PR is affected, please view them below:

Upgrade MacOS runner to 14

:x: 7 New Failures, 12 Unrelated Failures

As of commit 70864c88c61e93bd38a4985acfb293f8cf9a51ab with merge base d59b8105cd7437cccb7c480f15ce6caf67f587c6 ():

NEW FAILURES - The following jobs have failed:

* [Continuous Benchmark (PR) / CPU Pytest benchmark](https://hud.pytorch.org/pr/pytorch/rl/2188#25645337515) ([gh](https://github.com/pytorch/rl/actions/runs/9315521302/job/25645337515)) `Workflow failed! Resource not accessible by integration` * [Continuous Benchmark (PR) / GPU Pytest benchmark](https://hud.pytorch.org/pr/pytorch/rl/2188#25645338105) ([gh](https://github.com/pytorch/rl/actions/runs/9315521302/job/25645338105)) `Workflow failed! Resource not accessible by integration` * [Generate documentation / build-docs (3.9, 12.1) / linux-job](https://hud.pytorch.org/pr/pytorch/rl/2188#25645340571) ([gh](https://github.com/pytorch/rl/actions/runs/9315521346/job/25645340571)) * [Lint / c-source / linux-job](https://hud.pytorch.org/pr/pytorch/rl/2188#25645340310) ([gh](https://github.com/pytorch/rl/actions/runs/9315521345/job/25645340310)) * [Lint / python-source-and-configs / linux-job](https://hud.pytorch.org/pr/pytorch/rl/2188#25645341204) ([gh](https://github.com/pytorch/rl/actions/runs/9315521345/job/25645341204)) * [Unit-tests on Linux / tests-optdeps (3.10, 12.1) / linux-job](https://hud.pytorch.org/pr/pytorch/rl/2188#25645342749) ([gh](https://github.com/pytorch/rl/actions/runs/9315521342/job/25645342749)) `An error occurred trying to start process '/usr/bin/bash' with working directory '/home/ec2-user/actions-runner/_work/rl/rl/pytorch/rl'. No such file or directory` * [Unit-tests on Windows / unittests-cpu / windows-job](https://hud.pytorch.org/pr/pytorch/rl/2188#25645340011) ([gh](https://github.com/pytorch/rl/actions/runs/9315521343/job/25645340011)) `The process 'C:\Program Files\Git\cmd\git.exe' failed with exit code 128`

FLAKY - The following jobs failed but were likely due to flakiness present on trunk:

* [Examples Tests on Linux / tests (3.9, 12.1) / linux-job](https://hud.pytorch.org/pr/pytorch/rl/2188#25645339915) ([gh](https://github.com/pytorch/rl/actions/runs/9315521344/job/25645339915)) (matched **linux** rule in [flaky-rules.json](https://github.com/pytorch/test-infra/blob/generated-stats/stats/flaky-rules.json)) `The process '/usr/bin/git' failed with exit code 128` * [Habitat Tests on Linux / tests (3.9, 12.1) / linux-job](https://hud.pytorch.org/pr/pytorch/rl/2188#25645338756) ([gh](https://github.com/pytorch/rl/actions/runs/9315521334/job/25645338756)) (matched **linux** rule in [flaky-rules.json](https://github.com/pytorch/test-infra/blob/generated-stats/stats/flaky-rules.json)) `The process '/usr/bin/git' failed with exit code 128` * [Libs Tests on Linux / unittests-gym (3.9, 12.1) / linux-job](https://hud.pytorch.org/pr/pytorch/rl/2188#25645345229) ([gh](https://github.com/pytorch/rl/actions/runs/9315521336/job/25645345229)) (matched **linux** rule in [flaky-rules.json](https://github.com/pytorch/test-infra/blob/generated-stats/stats/flaky-rules.json)) `The process '/usr/bin/git' failed with exit code 128` * [Libs Tests on Linux / unittests-sklearn (3.9, 12.1) / linux-job](https://hud.pytorch.org/pr/pytorch/rl/2188#25645345563) ([gh](https://github.com/pytorch/rl/actions/runs/9315521336/job/25645345563)) (matched **linux** rule in [flaky-rules.json](https://github.com/pytorch/test-infra/blob/generated-stats/stats/flaky-rules.json)) `The process '/usr/bin/git' failed with exit code 128` * [RLHF Tests on Linux / unittests (3.9, 12.1) / linux-job](https://hud.pytorch.org/pr/pytorch/rl/2188#25645338935) ([gh](https://github.com/pytorch/rl/actions/runs/9315521341/job/25645338935)) (matched **linux** rule in [flaky-rules.json](https://github.com/pytorch/test-infra/blob/generated-stats/stats/flaky-rules.json)) `The process '/usr/bin/git' failed with exit code 128` * [Unit-tests on Linux / tests-cpu (3.10) / linux-job](https://hud.pytorch.org/pr/pytorch/rl/2188#25645340238) ([gh](https://github.com/pytorch/rl/actions/runs/9315521342/job/25645340238)) (matched **linux** rule in [flaky-rules.json](https://github.com/pytorch/test-infra/blob/generated-stats/stats/flaky-rules.json)) `The process '/usr/bin/git' failed with exit code 128` * [Unit-tests on Linux / tests-cpu (3.11) / linux-job](https://hud.pytorch.org/pr/pytorch/rl/2188#25645340943) ([gh](https://github.com/pytorch/rl/actions/runs/9315521342/job/25645340943)) (matched **linux** rule in [flaky-rules.json](https://github.com/pytorch/test-infra/blob/generated-stats/stats/flaky-rules.json)) `The process '/usr/bin/git' failed with exit code 128` * [Unit-tests on Linux / tests-cpu (3.8) / linux-job](https://hud.pytorch.org/pr/pytorch/rl/2188#25645341474) ([gh](https://github.com/pytorch/rl/actions/runs/9315521342/job/25645341474)) (matched **linux** rule in [flaky-rules.json](https://github.com/pytorch/test-infra/blob/generated-stats/stats/flaky-rules.json)) `The process '/usr/bin/git' failed with exit code 128` * [Unit-tests on Linux / tests-cpu (3.9) / linux-job](https://hud.pytorch.org/pr/pytorch/rl/2188#25645341675) ([gh](https://github.com/pytorch/rl/actions/runs/9315521342/job/25645341675)) (matched **linux** rule in [flaky-rules.json](https://github.com/pytorch/test-infra/blob/generated-stats/stats/flaky-rules.json)) `The process '/usr/bin/git' failed with exit code 128` * [Unit-tests on Linux / tests-gpu (3.10, 12.1) / linux-job](https://hud.pytorch.org/pr/pytorch/rl/2188#25645342065) ([gh](https://github.com/pytorch/rl/actions/runs/9315521342/job/25645342065)) (matched **linux** rule in [flaky-rules.json](https://github.com/pytorch/test-infra/blob/generated-stats/stats/flaky-rules.json)) `The process '/usr/bin/git' failed with exit code 128` * [Unit-tests on Linux / tests-olddeps (3.8, 11.6) / linux-job](https://hud.pytorch.org/pr/pytorch/rl/2188#25645342576) ([gh](https://github.com/pytorch/rl/actions/runs/9315521342/job/25645342576)) (matched **linux** rule in [flaky-rules.json](https://github.com/pytorch/test-infra/blob/generated-stats/stats/flaky-rules.json)) `The process '/usr/bin/git' failed with exit code 128` * [Unit-tests on Linux / tests-stable-gpu (3.10, 11.8) / linux-job](https://hud.pytorch.org/pr/pytorch/rl/2188#25645342914) ([gh](https://github.com/pytorch/rl/actions/runs/9315521342/job/25645342914)) (matched **linux** rule in [flaky-rules.json](https://github.com/pytorch/test-infra/blob/generated-stats/stats/flaky-rules.json)) `The process '/usr/bin/git' failed with exit code 128`

This comment was automatically generated by Dr. CI and updates every 15 minutes.

pytorch / rl