Skip to content

natten: add gb300 aarch64 wheel to v1.5.0 index #58

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
lfengad merged 2 commits into from
Jun 17, 2026
Merged

natten: add gb300 aarch64 wheel to v1.5.0 index #58

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
45be704
Build NATTEN for sm_103 (Blackwell Ultra / GB300)
pengcuo Jun 16, 2026
88c5dac
natten: add gb300 aarch64 wheel to v1.5.0 index
pengcuo Jun 16, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 4 additions & 1 deletion bin/_build.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -71,4 +71,7 @@ popd || exit 1
ccache --show-stats

# Fix wheel filenames.
uv run bin/fix_wheel.py -i "${OUTPUT_DIR}"/*.whl --version="${PACKAGE_VERSION}" --local-version="cu${CUDA_NAME}.torch${TORCH_NAME}"
# Optionally append a build-variant suffix to the local version (e.g. LOCAL_VERSION_SUFFIX=gb300
# -> '...+cu130.torch210.gb300') to distinguish a custom build from the upstream wheel.
LOCAL_VERSION="cu${CUDA_NAME}.torch${TORCH_NAME}${LOCAL_VERSION_SUFFIX:+.${LOCAL_VERSION_SUFFIX}}"
uv run bin/fix_wheel.py -i "${OUTPUT_DIR}"/*.whl --version="${PACKAGE_VERSION}" --local-version="${LOCAL_VERSION}"
1 change: 1 addition & 0 deletions bin/build.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -62,6 +62,7 @@ env -i \
PACKAGE_VERSION="${PACKAGE_VERSION}" \
PYTHON_VERSION="${PYTHON_VERSION}" \
TORCH_VERSION="${TORCH_VERSION}" \
LOCAL_VERSION_SUFFIX="${LOCAL_VERSION_SUFFIX:-}" \
OUTPUT_NAME="${OUTPUT_NAME}" \
OUTPUT_DIR="${OUTPUT_DIR}" \
PATH="${PATH:-}" \
Expand Down
1 change: 1 addition & 0 deletions docs/v1.5.0/natten/index.html
View file
Open in desktop

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

5 changes: 4 additions & 1 deletion packages/natten/build.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,10 @@ esac
# https://natten.org/install/#build-natten-libnatten
export NATTEN_N_WORKERS=${NATTEN_N_WORKERS:-$(($(nproc) / 2))}
export NATTEN_VERBOSE=1
export NATTEN_CUDA_ARCH="${TORCH_CUDA_ARCH_LIST}"
# Append sm_103 (Blackwell Ultra / GB300): not in PyTorch's prebuilt arch list,
# so TORCH_CUDA_ARCH_LIST omits it and NATTEN's Blackwell kernels fall back to
# "no kernel image" on GB300. NATTEN maps 10.3 -> sm_103a automatically.
export NATTEN_CUDA_ARCH="${TORCH_CUDA_ARCH_LIST};10.3"

pip wheel \
-v \
Expand Down
Loading