transformer-engine-cu12 PyPI wheels have incorrect WHEEL tag (cp310 hardcoded), causing reinstall loops

[leoleoasd]

Describe the bug

The transformer-engine-cu12 wheels published to PyPI have a mismatch between the wheel filename tag and the internal WHEEL metadata tag.

Filename: transformer_engine_cu12-X.Y.Z-py3-none-manylinux_2_28_x86_64.whl (correct — Python-version agnostic)

Internal WHEEL file: Tag: cp310-cp310-manylinux_2_28_x86_64 (incorrect — hardcoded to Python 3.10)

This affects every recent release tested:

Version	Filename Tag	Internal WHEEL Tag
2.10.0	py3-none-manylinux_2_28_x86_64	cp310-cp310-manylinux_2_28_x86_64
2.12.0	py3-none-manylinux_2_28_x86_64	cp310-cp310-manylinux_2_28_x86_64
2.13.0	py3-none-manylinux_2_28_x86_64	cp310-cp310-manylinux_2_28_x86_64

For comparison, the sibling packages have correct tags:

• transformer-engine (meta): py3-none-any (correct)
• transformer-engine-torch: cp312-cp312-linux_x86_64 (correct, matches build Python)

Reproduction

# On Python 3.12
pip download transformer-engine-cu12==2.13.0 --no-deps -d /tmp/te
unzip -p /tmp/te/transformer_engine_cu12-2.13.0-py3-none-manylinux_2_28_x86_64.whl \
  transformer_engine_cu12-2.13.0.dist-info/WHEEL

Output:

Wheel-Version: 1.0
Generator: setuptools (82.0.1)
Root-Is-Purelib: false
Tag: cp310-cp310-manylinux_2_28_x86_64

Root Cause

The release build environment likely uses Python 3.10, and setuptools records the build Python's tag in the WHEEL metadata. Since transformer-engine-cu12 bundles precompiled .so files and is not ABI-dependent on a specific Python version, the wheel filename is correctly overridden to py3-none, but the internal WHEEL metadata file is not updated to match.

Expected Behavior

The internal WHEEL tag should match the filename tag: py3-none-manylinux_2_28_x86_64 (since the package is Python-version agnostic).

Environment

• Python: 3.12
• uv: 0.11.6
• OS: Linux x86_64

[leoleoasd]

AI suggested this. @ksivaman can you take a look?

Suggested Fix

The fix is a one-line addition to build_tools/wheel_utils/build_wheels.sh. Between lines 46 and 47 (after renaming the dist-info directory but before wheel pack), add a sed to rewrite the Tag: line in the WHEEL metadata to match the intended py3-none tag:

sed -i "s/Tag: cp310-cp310/Tag: py3-none/g" "${WHL_BASE}/transformer_engine_cu${CUDA_MAJOR}-${VERSION}.dist-info/WHEEL"

This is consistent with the existing approach — the script already uses sed to patch the METADATA file in the same unpack/repack cycle. The same fix should also be applied to the equivalent lines in qa/L0_pytorch_wheel/test.sh and qa/L0_jax_wheel/test.sh.

Alternatively, wheel tags --python-tag py3 --abi-tag none can be run on the final .whl file to atomically rewrite both the filename and internal WHEEL metadata, which would also eliminate the manual filename rename on lines 50-54.

[ptrendx]

@ksivaman Please take a look at this.