Remove workaround to old CUDA bug

[pearu]

Stack from ghstack (oldest at bottom):

• (to be filled)

Pull-Request-resolved: #164354

cc @jgong5 @mingfeima @XiaobingSuper @sanchitintel @ashokei @jingxu10 @jerryzh168

[pytorch-bot]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR
• 📄 Preview C++ docs built from this PR
• ❓ Need help or want to give feedback on the CI? Visit the bot commands wiki

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

❌ 2 New Failures, 22 Cancelled Jobs, 1 Pending

As of commit 04a2524 with merge base b5c4f46 ():

NEW FAILURES - The following jobs have failed:

• Check Labels / Check labels (gh)
  RuntimeError: Error checking labels: PR does not have required labels
• Check mergeability of ghstack PR / ghstack-mergeability-check (gh)
  RuntimeError: Found an unexpected number of PRs mentioned in commit c31a851ef9cf3ea4a445ce2db713fec3bf449f73: 2. This is probably because you are using an old version of ghstack. Please update ghstack and resubmit your PRs

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

• Lint / Link checks / lint-urls / lint-urls (gh)
• Lint / Link checks / lint-xrefs / lint-xrefs (gh)
• Lint / lintrunner-clang / linux-job (gh)
  A task was canceled.
• Lint / lintrunner-noclang / linux-job (gh)
• Lint / pr-sanity-checks (gh)
• Lint / quick-checks / linux-job (gh)
• Lint / Test tools / linux-job (gh)
• Lint / toc / linux-job (gh)
• Lint / workflow-checks / linux-job (gh)
• pull / before-test / target-determination (gh)
• pull / cuda12.8-py3.10-gcc9-sm75 / build (gh)
• pull / linux-jammy-cuda12.8-cudnn9-py3.10-clang12 / build (gh)
• pull / linux-jammy-py3.10-clang12 / build (gh)
• pull / linux-jammy-py3.10-clang12-onnx / build (gh)
• pull / linux-jammy-py3.10-clang18-asan / build (gh)
  ##[error]The operation was canceled.
• pull / linux-jammy-py3.10-gcc11 / build (gh)
• pull / linux-jammy-py3.10-gcc11-mobile-lightweight-dispatch-build / build (gh)
• pull / linux-jammy-py3.10-gcc11-no-ops / build (gh)
• pull / linux-jammy-py3.10-gcc11-pch / build (gh)
• pull / linux-jammy-py3.13-clang12 / build (gh)
• pull / linux-jammy-rocm-py3.10 / build (gh)
• pull / linux-jammy-xpu-n-py3.10 / build (gh)

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

[github-actions]

This PR needs a release notes: label

If your changes are user facing and intended to be a part of release notes, please use a label starting with release notes:.

If not, please add the topic: not user facing label.

To add a label, you can comment to pytorchbot, for example
@pytorchbot label "topic: not user facing"

For more information, see
https://github.com/pytorch/pytorch/wiki/PyTorch-AutoLabel-Bot#why-categorize-for-release-notes-and-how-does-it-work.