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: Python 3.15 support for PyTorch #184352
• Skip PR: Linux py 3.15 wheel builds #184600
• Related downstream: Publish Python 3.15 packages (target release 2026-10-01) 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