[Feature] DT compatibility with compile

[vmoens]

Stack from ghstack (oldest at bottom):

• [Feature] GAIL compatibility with compile #2573
• [Feature] DQN compatibility with compile #2571
• [Feature] Discrete SAC compatibility with compile #2569
• -> [Feature] DT compatibility with compile #2556
• [Feature] DDPG compatibility with compile #2555
• [Feature] CROSSQ compatibility with compile #2554
• [Feature] CQL compatibility with compile #2553

[pytorch-bot]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR

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

❗ 1 Active SEVs

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

• [DomainsOnly] Jobs fail with GLIBC version not found

❌ 15 New Failures, 2 Pending, 4 Unrelated Failures

As of commit ab214d4 with merge base b4b5944 ():

NEW FAILURES - The following jobs have failed:

• Continuous Benchmark (PR) / CPU Pytest benchmark (gh)
  FAILED ../../../../../../tmp/test_objectives_benchmarks.py::test_iql_speed[reduce-overhead-backward] - torch._dynamo.exc.Unsupported: Graph break due to unsupported builtin _C.expand. This function is either a Python builtin (e.g. _warnings.warn) or a third-party C/C++ Python extension (perhaps created with pybind). If it is a Python builtin, please file an issue on GitHub so the PyTorch team can add support for it and see the next case for a workaround. If it is a third-party C/C++ Python extension, please either wrap it into a PyTorch-understood custom operator (see https://pytorch.org/tutorials/advanced/custom_ops_landing_page.html for more details) or, if it is traceable, use torch.compiler.allow_in_graph.
• Continuous Benchmark (PR) / GPU Pytest benchmark (gh)
  FAILED ../../../../tmp/test_objectives_benchmarks.py::test_iql_speed[reduce-overhead-backward] - torch._dynamo.exc.Unsupported: Graph break due to unsupported builtin _C.expand. This function is either a Python builtin (e.g. _warnings.warn) or a third-party C/C++ Python extension (perhaps created with pybind). If it is a Python builtin, please file an issue on GitHub so the PyTorch team can add support for it and see the next case for a workaround. If it is a third-party C/C++ Python extension, please either wrap it into a PyTorch-understood custom operator (see https://pytorch.org/tutorials/advanced/custom_ops_landing_page.html for more details) or, if it is traceable, use torch.compiler.allow_in_graph.
• Libs Tests on Linux / unittests-gym (3.9, 12.1) / linux-job (gh)
  test/test_libs.py::TestGym::test_gym_gymnasium_parallel
• Libs Tests on Linux / unittests-sklearn (3.9, 12.1) / linux-job (gh)
  test/test_libs.py::TestOpenML::test_env[batch_size2-magic]
• Lint / python-source-and-configs / linux-job (gh)
  RuntimeError: Command docker exec -t e3ffe1fce9b86bb1782fb2a0ccf6d420d3e1f22d24f485420eda19f284614a87 /exec failed with exit code 1
• SOTA Tests on Linux / tests (3.9, 12.1) / linux-job (gh)
  AttributeError: 'TransformedEnv' object has no attribute 'single_action_spec'
• Unit-tests on Linux / tests-cpu (3.10) / linux-job (gh)
  test/test_transforms.py::TestActionDiscretizer::test_transform_env[SamplingStrategy.RANDOM-False-True-pendulum]
• Unit-tests on Linux / tests-cpu (3.11) / linux-job (gh)
  test/test_transforms.py::TestActionDiscretizer::test_transform_env[3-False-True-pendulum]
• Unit-tests on Linux / tests-cpu (3.12) / linux-job (gh)
  test/test_transforms.py::TestActionDiscretizer::test_transform_env[3-False-True-pendulum]
• Unit-tests on Linux / tests-cpu (3.9) / linux-job (gh)
  test/test_transforms.py::TestActionDiscretizer::test_transform_env[SamplingStrategy.RANDOM-False-True-pendulum]
• Unit-tests on Linux / tests-cpu-oldget (3.12) / linux-job (gh)
  test/test_transforms.py::TestActionDiscretizer::test_transform_env[3-False-True-pendulum]
• Unit-tests on Linux / tests-gpu (3.11, 12.1) / linux-job (gh)
  test/test_transforms.py::TestActionDiscretizer::test_transform_env[3-False-True-pendulum]
• Unit-tests on Linux / tests-olddeps (3.8, 11.6) / linux-job (gh)
  test/test_transforms.py::TestActionDiscretizer::test_transform_env[SamplingStrategy.RANDOM-False-True-pendulum]
• Unit-tests on Linux / tests-optdeps (3.11, 12.1) / linux-job (gh)
  test/test_transforms.py::TestActionDiscretizer::test_transform_compose
• Unit-tests on Linux / tests-stable-gpu (3.10, 11.8) / linux-job (gh)
  test/test_transforms.py::TestActionDiscretizer::test_transform_env[SamplingStrategy.RANDOM-False-True-pendulum]

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

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

• Build Linux Wheels / pytorch/rl (pytorch/rl, test/smoke_test.py, torchrl) / upload / manywheel-py3_9-rocm6_1 (gh) (trunk failure)
  ##[error]Unable to find an artifact with the name: pytorch_rl__3.9_rocm6.1_x86_64
• Build Linux Wheels / pytorch/rl (pytorch/rl, test/smoke_test.py, torchrl) / upload / manywheel-py3_9-rocm6_2 (gh) (trunk failure)
  ##[error]Unable to find an artifact with the name: pytorch_rl__3.9_rocm6.2_x86_64
• Habitat Tests on Linux / tests (3.9, 12.1) / linux-job (gh) (trunk failure)
  AttributeError: _ARRAY_API not found
• Unit-tests on Windows / unittests-cpu / windows-job (gh) (trunk failure)
  ##[error]Process completed with exit code 1.

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