[Feature] Add _Contiguous module and reshape improvements to encoders/decoders

[vmoens]

Stack from ghstack (oldest at bottom):

• -> [Feature] Add _Contiguous module and reshape improvements to encoders/decoders #3306
• [BugFix] Fix SliceSampler for torch.compile compatibility #3298

---

• Add _Contiguous helper module for torch.compile inductor compatibility
• Refactor ObsEncoder.forward and ObsDecoder.forward to use flatten/unflatten with contiguous()
• Add _maybe_record_function_decorator for profiling

[pytorch-bot]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR

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

❌ 13 New Failures, 2 Cancelled Jobs, 2 Unrelated Failures

As of commit 348c94d with merge base df00d61 ():

NEW FAILURES - The following jobs have failed:

• Continuous Benchmark (PR) / CPU Pytest benchmark (gh)
  Process completed with exit code 1.
• Habitat Tests on Linux / tests (3.9, 12.8) / linux-job (gh)
  RuntimeError: Command docker exec -t cd1439ad8881b565a23f7b52e1037471b430f2f76d404680c964b6a0a2b2d0b2 /exec failed with exit code 1
• Libs Tests on Linux / unittests-gym (3.9, 12.8) / linux-job (gh)
  RuntimeError: Command docker exec -t 727ca317c5b7a32bad25900879bdb35d99a98ab202ee19473cd4759165946bde /exec failed with exit code 1
• Unit-tests on Linux / tests-cpu (3.10) / linux-job (gh)
  test/test_transforms.py::TestRayModuleTransform::test_ray_module_transform_scheme_collector
• Unit-tests on Linux / tests-cpu (3.11) / linux-job (gh)
  test/test_transforms.py::TestRayModuleTransform::test_ray_module_transform_scheme_collector
• Unit-tests on Linux / tests-cpu (3.12) / linux-job (gh)
  test/test_transforms.py::TestRayModuleTransform::test_ray_module_transform_scheme_collector
• Unit-tests on Linux / tests-cpu (3.13) / linux-job (gh)
  test/test_transforms.py::TestRayModuleTransform::test_ray_module_transform_scheme_collector
• Unit-tests on Linux / tests-gpu (3.12, 13.0) / linux-job (gh)
  test/test_transforms.py::TestRayModuleTransform::test_ray_module_transform_scheme_collector
• Unit-tests on Linux / tests-gpu-distributed (3.12, 13.0) / linux-job (gh)
  test/test_distributed.py::TestRayCollector::test_ray_collector_policy_constructor
• Unit-tests on Linux / tests-olddeps (3.9, 11.6) / linux-job (gh)
  test/test_transforms.py::TestRayModuleTransform::test_ray_module_transform_scheme_collector
• Unit-tests on Linux / tests-optdeps (3.12, 13.0) / linux-job (gh)
  test/test_transforms.py::TestRayModuleTransform::test_ray_module_transform_scheme_collector
• Unit-tests on Linux / tests-stable-gpu (3.12, 13.0) / linux-job (gh)
  test/test_transforms.py::TestRayModuleTransform::test_ray_module_transform_scheme_collector
• Unit-tests on Linux / tests-stable-gpu-distributed (3.12, 13.0) / linux-job (gh)
  test/test_distributed.py::TestRayCollector::test_ray_collector_policy_constructor

CANCELLED JOBS - The following jobs were cancelled. Please retry:

• Continuous Benchmark (PR) / GPU Pytest benchmark (gh)
• SOTA Tests on Linux / tests (3.10, 13.0) / linux-job (gh)
  ##[error]The operation was canceled.

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

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

• Unit-tests on Linux / tests-cpu (3.14) / linux-job (gh) (trunk failure)
  test/test_transforms.py::TestConditionalSkip::test_trans_parallel_env_check[False-False]
• Unit-tests on Windows / unittests-cpu (3.10, windows.4xlarge, cpu) / windows-job (gh) (trunk failure)
  test/test_transforms.py::TestConditionalSkip::test_trans_parallel_env_check[False-False]

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