Issue Station
Sign In
Tools hereToolbox

pytorch - 💡(How to fix) Fix ROCm: Python 3.15 wheel builds blocked by composable_kernel using removed importlib.abc.Loader.load_module()

Official PRs (…)
ON THIS PAGE

Recommended Tools

×6

Utilities matched from this issue’s tags and category — try them while you read without losing context.

GitHub issue graph ai analysis

Paste a GitHub issue URL. We fetch that issue, discover linked issues from bodies/comments/timeline, collect linked pull requests, and produce a structured English report.

The report is written in English Markdown for sharing and archival.

Helpful · Quick feedback

Loading…

Python 3.15 / 3.15t support is being added across the PyTorch build matrix (umbrella: #184352). PR #184600 enabled 3.15 and 3.15t Linux manywheel builds, but skipped ROCm because the ROCm path fails to build on Python 3.15.

Root Cause

composable_kernel's ck_tile/01_fmha/generate.py still uses importlib.abc.Loader.load_module(), which was removed in Python 3.15 (deprecated since 3.4).

The skip was added in .github/scripts/generate_binary_build_matrix.py in #184600:

# TODO: Re-enable ROCm for python 3.15 once composable_kernel's
# ck_tile/01_fmha/generate.py is updated; it still uses
# importlib.abc.Loader.load_module() which was removed in 3.15.
if arch_version in ROCM_ARCHES and (
    python_version == "3.15" or python_version == "3.15t"
):
    continue

Code Example

# TODO: Re-enable ROCm for python 3.15 once composable_kernel's
# ck_tile/01_fmha/generate.py is updated; it still uses
# importlib.abc.Loader.load_module() which was removed in 3.15.
if arch_version in ROCM_ARCHES and (
    python_version == "3.15" or python_version == "3.15t"
):
    continue
RAW_BUFFERClick to expand / collapse

Tracking issue for the AMD/ROCm team to unblock Python 3.15 wheel builds.

Context

Python 3.15 / 3.15t support is being added across the PyTorch build matrix (umbrella: #184352). PR #184600 enabled 3.15 and 3.15t Linux manywheel builds, but skipped ROCm because the ROCm path fails to build on Python 3.15.

Root cause

composable_kernel's ck_tile/01_fmha/generate.py still uses importlib.abc.Loader.load_module(), which was removed in Python 3.15 (deprecated since 3.4).

The skip was added in .github/scripts/generate_binary_build_matrix.py in #184600:

# TODO: Re-enable ROCm for python 3.15 once composable_kernel's
# ck_tile/01_fmha/generate.py is updated; it still uses
# importlib.abc.Loader.load_module() which was removed in 3.15.
if arch_version in ROCM_ARCHES and (
    python_version == "3.15" or python_version == "3.15t"
):
    continue

Action items

  • Update composable_kernel (ck_tile/01_fmha/generate.py) to use importlib.util.spec_from_file_location / module_from_spec (or equivalent) instead of Loader.load_module().
  • Pull the upstream fix into PyTorch.
  • Remove the skip block in .github/scripts/generate_binary_build_matrix.py and re-enable ROCm wheel builds for 3.15 / 3.15t.

References

  • Umbrella tracking: #184352
  • Skip PR: #184600
  • Related downstream: ROCm/TheRock#5400

cc @malfet @jeffdaily @sunway513 @jithunnair-amd @pruthvistony @ROCmSupport @jataylo @hongxiayang @naromero77amd @pragupta @jerrymannil @xinyazhang @pytorch/pytorch-dev-infra

Suggested labels: module: rocm, module: ci, module: python version

Vote matrix · Quick signals

Works
Did the solution work? Tap to confirm.
Easy Fix
Was it a quick fix?
Time Saver
Did it save you time?
Blocking
Was it severely blocking?
Common Issue
Are others likely hitting this too?
Flaky / Intermittent
Is it intermittent?
Verified / Reproducible
Can you reproduce it reliably?
Loading…

Still need to ship something?

×6

Another batch ranked right after the header list — different links, same matching logic.

Back to top recommendations

TRENDING

pytorch - 💡(How to fix) Fix ROCm: Python 3.15 wheel builds blocked by composable_kernel using removed importlib.abc.Loader.load_module()