Skip to content

[Build] Restore CUDA 13 Linux aarch64 core wheels #3309

Description

@bvolpato

Describe the bug

transformer-engine-cu13 stopped publishing Linux aarch64 wheels after v2.16.0.

PyPI contains both platforms for v2.16.0:

  • transformer_engine_cu13-2.16.0-py3-none-manylinux_2_28_aarch64.whl
  • transformer_engine_cu13-2.16.0-py3-none-manylinux_2_28_x86_64.whl

For v2.16.1 and v2.17.0, PyPI contains only the x86_64 wheel. This makes installation on GB200/aarch64 fall back to a resource-intensive native build.

Steps to reproduce

Inspect published files for:

Or attempt an ARM64 binary-only download:

python -m pip download \
  --only-binary=:all: \
  --platform manylinux_2_28_aarch64 \
  --python-version 3.12 \
  --implementation cp \
  --abi cp312 \
  --no-deps \
  transformer-engine-cu13==2.17.0

Expected behavior

Publish transformer_engine_cu13-<version>-py3-none-manylinux_2_28_aarch64.whl alongside the x86_64 core wheel for CUDA 13 releases.

ARM64 release automation also appears unhealthy:

The v2.16.1 runner reported No space left on device while writing its diagnostic log. This may be separate from the core-wheel publishing path, but it shows the release currently lacks reliable ARM64 coverage.

Environment overview

  • Platform: Linux aarch64
  • Hardware: NVIDIA GB200
  • Python: 3.12
  • CUDA: 13.0
  • Transformer Engine: 2.16.1 and 2.17.0

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions