[ethanwee1]

Problem

TheRock force-includes torch/_rocm_init.py into the ROCm wheel as a scikit-build-core workaround, but upstream PyTorch now installs that file via torch/CMakeLists.txt (guarded SKBUILD AND USE_ROCM), so the force-include is redundant.

Changes

Remove the torch/_rocm_init.py force-include from build_prod_wheels.py and update the comment. The Windows DLL force-includes stay until their upstream PRs land.

Validation

Windows ROCm wheel builds (gfx1100, py3.12) on this PR's head (523ac105f7) passed build and the import torch sanity check:

• release/2.13: https://github.com/ROCm/TheRock/actions/runs/38811386532 (setuptools backend ships _rocm_init.py as package data)
• release/2.14: https://github.com/ROCm/TheRock/actions/runs/38811396288
• nightly: https://github.com/ROCm/TheRock/actions/runs/38811415580

release/2.15 can't be built yet because ROCm/pytorch release/2.15 has no related_commits file (https://github.com/ROCm/TheRock/actions/runs/38811406174 fails at checkout). Merges independently of the other PRs below, and cleanly alongside #9058.

Related PR:

• [ROCm][Windows] Install libomp140.x86_64.dll into scikit-build ROCm wheels pytorch/pytorch#191629: install libomp140.x86_64.dll via CMake
• [ROCm][Windows] Install aotriton_v2.dll and liblzma.dll into scikit-build ROCm wheels pytorch/pytorch#200931: install aotriton_v2.dll + liblzma.dll via CMake
• [pytorch] Drop _rocm_init.py wheel force-include (now upstream) #9055: drop the _rocm_init.py wheel force-include (this PR)
• [pytorch] Drop aotriton_v2.dll + liblzma.dll wheel force-includes (now upstream) #9058: drop the aotriton_v2.dll + liblzma.dll wheel force-includes
• [pytorch] Drop libomp140.x86_64.dll wheel force-include (now upstream) #9360: drop the libomp140.x86_64.dll wheel force-include
• [release/2.14] Install Windows ROCm wheel DLLs: aotriton_v2, liblzma, libomp140 pytorch#3835: release/2.14 cherry-pick of [ROCm][Windows] Install libomp140.x86_64.dll into scikit-build ROCm wheels pytorch/pytorch#191629 + [ROCm][Windows] Install aotriton_v2.dll and liblzma.dll into scikit-build ROCm wheels pytorch/pytorch#200931
• [release/2.15] Install Windows ROCm wheel DLLs: aotriton_v2, liblzma, libomp140 pytorch#3836: release/2.15 cherry-pick of [ROCm][Windows] Install libomp140.x86_64.dll into scikit-build ROCm wheels pytorch/pytorch#191629 + [ROCm][Windows] Install aotriton_v2.dll and liblzma.dll into scikit-build ROCm wheels pytorch/pytorch#200931
• Issue: [Issue] PyTorch nightly builds are missing _rocm_init.py. Windows fails to load c10_hip.dll, Linux fails to load libMIOpen.so.1 #6965

Tracked in AIPYTORCH-1516.

[therock-pr-bot]

❌ PR Check — Action Required

Check	Status	Details
📝 PR Description	❌ Fail	Error: PR description must reference a JIRA ID, ISSUE ID, or a GitHub closing keyword. Expected: include a JIRA ID / ISSUE ID line (separator : or -, or omitted; value may be a JIRA key, a number with/without #, or a link), OR a closing keyword + issue reference. Accepted examples: • JIRA ID : TESTAUTO-6039 • JIRA ID - #330 • JIRA ID #330 • JIRA ID (on separate line) ROCM-25757 • ISSUE ID : TESTUTO-3334 • ISSUE ID #3334 • ISSUE ID - TESTAUTO-3433 • ISSUE ID (on separate line) AIRUNTIME-2352 • ISSUE ID : https://github.com/<org_name>/<repo_name>/issues/1234 • Closes #10 • Fixes octo-org/octo-repo#100 • Resolves: #123 • #123 • https://github.com/<org_name>/<repo_name>/issues/123 Current: no valid JIRA/ISSUE/closing-keyword reference found
⛔ Forbidden Files	✅ Pass	—
🧪 Unit Test	⚠️ Warning	Error: Source/code files changed without an accompanying unit test. Expected: add at least one test file named like test_<name>.py / test_<name>.cpp (or <name>_test.*). Current: code file(s) changed: external-builds/pytorch/build_prod_wheels.py; no test file found
🔎 pre-commit	⏳ Pending	⏳ Still running…
🚫 Draft PR	🔜 To Be Enabled	—
🚩 Feature Flag	🔜 To Be Enabled	—
📊 Code Coverage	🔜 To Be Enabled	—

⚠️ 1 policy check(s) failed. Please address the issues above before this PR can be Reviewed.

🚫 Please fix the failed policies

• ❌ PR Description

The Not ready to Review label was added to this PR. Once all policies pass, the label is removed automatically.

📖 Need help? See the Policy FAQ for details on every check and how to fix failures.

🙋 Wish to Override Policy?

[therock-pr-bot]

🚫 Please fix the failed policies before requesting reviews.

The following policy checks failed:

• ❌ PR Description

The Not ready to Review label has been added to this PR.
Once all policies pass, the label will be removed automatically.