[BugFix] Fix DeviceCastTransform

[vmoens]

Stack from ghstack (oldest at bottom):

• -> [BugFix] Fix DeviceCastTransform #2471

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/rl/2471

• 📄 Preview Python docs built from this PR

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

❌ 8 New Failures, 4 Unrelated Failures

As of commit 784d024 with merge base fac1f7a ():

NEW FAILURES - The following jobs have failed:

• Continuous Benchmark (PR) / CPU Pytest benchmark (gh)
  /home/runner/work/_temp/a9e80b72-0fde-40c1-89af-dd3aa2c9dfcb.sh: line 5: 2147 Bus error (core dumped) $RUN_BENCHMARK /tmp/tmp.9ah91mCeSJ
• Continuous Benchmark (PR) / GPU Pytest benchmark (gh)
  Fatal Python error: Fatal Python error: Fatal Python error: Fatal Python error: Bus errorBus error
• Habitat Tests on Linux / tests (3.9, 12.1) / linux-job (gh)
  RuntimeError: Command docker exec -t cc32baf3322b6e3f57fe7d5e29323ac4c97e76acd82ba91e2b750e6fe4ef69ef /exec failed with exit code 134
• Lint / c-source / linux-job (gh)
  RuntimeError: Command docker exec -t 9dbf60e6e7cc73f42f078922178c1555d526975c8a3cf9f16e18f71de793dff1 /exec failed with exit code 1
• Unit-tests on Linux / tests-cpu (3.12) / linux-job (gh)
  RuntimeError: Command docker exec -t b13fa530f2bec120f79ee39fe1007963c894f6339a0d60cbda20004bc5b24090 /exec failed with exit code 1
• Unit-tests on Linux / tests-cpu-oldget (3.12) / linux-job (gh)
  RuntimeError: Command docker exec -t 5f2ebc39242f062cf52a45ab2cc8dc7e2411f25a73aeb175d02426e09305c780 /exec failed with exit code 1
• Unit-tests on Linux / tests-olddeps (3.8, 11.6) / linux-job (gh)
  test/test_transforms.py::TestKLRewardTransform::test_kl_lstm
• Unit-tests on Windows / unittests-cpu / windows-job (gh)
  RuntimeError: Compiler: cl is not found.

BROKEN TRUNK - The following jobs failed but were present on the merge base:

👉 Rebase onto the `viable/strict` branch to avoid these failures

• Build Windows Wheels / pytorch/rl / upload / wheel-py3_9-cpu (gh) (trunk failure)
  ##[error]Unable to find any artifacts for the associated workflow
• Build Windows Wheels / pytorch/rl / upload / wheel-py3_9-cuda11_8 (gh) (trunk failure)
  ##[error]Unable to find any artifacts for the associated workflow
• Build Windows Wheels / pytorch/rl / upload / wheel-py3_9-cuda12_1 (gh) (trunk failure)
  ##[error]Unable to find any artifacts for the associated workflow
• Build Windows Wheels / pytorch/rl / upload / wheel-py3_9-cuda12_4 (gh) (trunk failure)
  ##[error]Unable to find any artifacts for the associated workflow

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