[BugFix] update_policy_weights_() with cudagraph

[vmoens]

Actually solved by pytorch/tensordict#1346

[pytorch-bot]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR

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

❌ 3 New Failures, 30 Pending

As of commit 849802c with merge base 16b70be ():

NEW FAILURES - The following jobs have failed:

• Lint / python-source-and-configs / linux-job (gh)
  RuntimeError: Command docker exec -t d566bc56194871963107058445aba78252e13e3443a843160bb2e66adaf911f8 /exec failed with exit code 1
• Push Binary Nightly / build-wheel-windows (3.13, 3.13) (gh)
• Unit-tests on Windows / unittests-cpu (3.10, windows.4xlarge, cpu) / windows-job (gh)
  Process completed with exit code 2.

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