Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
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
2 changes: 2 additions & 0 deletions benchmark/nixlbench/contrib/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -188,6 +188,8 @@ RUN uv pip install --upgrade meson meson-python pybind11 patchelf pyYAML click t
RUN CUDA_SHORT_VERSION=cu$(echo $CUDA_VERSION | cut -d. -f1,2 | tr -d .) && \
FLAGS="--index-url https://download.pytorch.org/whl/$CUDA_SHORT_VERSION" && \
uv pip install $FLAGS torch torchvision torchaudio
# Upgrade setuptools to latest version for compatibility with PEP 639 (license format)
RUN uv pip install --upgrade 'setuptools>=80.9.0'

COPY --from=nixl . /workspace/nixl
COPY --from=nixlbench . /workspace/nixlbench
Expand Down
2 changes: 2 additions & 0 deletions contrib/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -190,6 +190,8 @@ RUN uv pip install --upgrade meson meson-python pybind11 patchelf pyYAML click t
# Install PyTorch
RUN export UV_INDEX="https://download.pytorch.org/whl/cu$(echo $CUDA_VERSION | cut -d. -f1,2 | tr -d .)" && \
uv pip install torch torchvision torchaudio
# Upgrade setuptools to latest version for compatibility with PEP 639 (license format)
RUN uv pip install --upgrade 'setuptools>=80.9.0'

WORKDIR /workspace/nixl
COPY . /workspace/nixl
Expand Down
7 changes: 5 additions & 2 deletions contrib/Dockerfile.manylinux
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ ARG BASE_IMAGE
ARG BASE_IMAGE_TAG
FROM ${BASE_IMAGE}:${BASE_IMAGE_TAG}

ARG DEFAULT_PYTHON_VERSION="3.12"
ARG DEFAULT_PYTHON_VERSION="3.14"
ARG ARCH="x86_64"
ARG UCX_REF="v1.20.x"
ARG LIBFABRIC_VERSION="v1.21.0"
Expand Down Expand Up @@ -243,6 +243,8 @@ RUN uv pip install --upgrade meson meson-python pybind11 patchelf pyYAML click s
# Install PyTorch
RUN export UV_INDEX="https://download.pytorch.org/whl/cu$(echo $CUDA_VERSION | cut -d. -f1,2 | tr -d .)" && \
uv pip install torch torchvision torchaudio
# Upgrade setuptools to latest version for compatibility with PEP 639 (license format)
RUN uv pip install --upgrade 'setuptools>=80.9.0'

COPY . /workspace/nixl
WORKDIR /workspace/nixl
Expand All @@ -266,10 +268,11 @@ RUN echo "/usr/local/nixl/lib/$ARCH-linux-gnu" > /etc/ld.so.conf.d/nixl.conf &&

# Create the wheel
# No need to specifically add path to libcuda.so here, meson finds the stubs and links them
ARG WHL_PYTHON_VERSIONS="3.9,3.10,3.11,3.12,3.13"
ARG WHL_PYTHON_VERSIONS="3.10,3.11,3.12,3.13,3.14"
ARG WHL_PLATFORM="manylinux_2_28_$ARCH"
RUN IFS=',' read -ra PYTHON_VERSIONS <<< "$WHL_PYTHON_VERSIONS" && \
export UV_INDEX="https://download.pytorch.org/whl/cu$(echo $CUDA_VERSION | cut -d. -f1,2 | tr -d .)" && \
export UV_INDEX_STRATEGY=unsafe-best-match && \
mkdir -p dist && \
for PYTHON_VERSION in "${PYTHON_VERSIONS[@]}"; do \
export PATH=$VIRTUAL_ENV/bin:$PATH && \
Expand Down
19 changes: 10 additions & 9 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -14,18 +14,19 @@
# limitations under the License.

[build-system]
requires = ["meson-python", "pybind11", "patchelf", "pyyaml", "types-PyYAML", "pytest", "build", "setuptools"]
requires = ["meson-python", "pybind11", "patchelf", "pyyaml", "types-PyYAML", "pytest", "build", "setuptools>=80.9.0"]
build-backend = "mesonpy"

[project]
name = 'nixl-cu12'
version = '0.8.0'
description = 'NIXL Python API'
readme = 'README.md'
license = {file = 'LICENSE'}
requires-python = '>=3.9'
name = "nixl-cu12"
version = "0.8.0"
description = "NIXL Python API"
readme = "README.md"
license = "Apache-2.0"
license-files = ["LICENSE"]
requires-python = ">=3.10"
authors = [
{name = 'NIXL Developers', email = '[email protected]'}
{name = "NIXL Developers", email = "[email protected]"}
]

dependencies = ["torch", "numpy"]
Expand All @@ -34,4 +35,4 @@ dependencies = ["torch", "numpy"]
profile = "black"

[tool.meson-python.args]
setup = ['-Dinstall_headers=false']
setup = ["-Dinstall_headers=false"]
9 changes: 5 additions & 4 deletions src/bindings/python/nixl-meta/pyproject.toml.in
Original file line number Diff line number Diff line change
Expand Up @@ -14,18 +14,19 @@
# limitations under the License.

[build-system]
requires = ["setuptools"]
requires = ["setuptools>=80.9.0"]
build-backend = "setuptools.build_meta"

[project]
name = "nixl"
version = "@VERSION@"
description = "NIXL Python API meta package for CUDA variants"
readme = "README.md"
license = {file = 'LICENSE'}
requires-python = ">=3.9"
license = "Apache-2.0"
license-files = ["LICENSE"]
requires-python = ">=3.10"
authors = [
{name = 'NIXL Developers', email = '[email protected]'}
{name = "NIXL Developers", email = "[email protected]"}
]

dependencies = ["@WHEEL_DEP@>=@VERSION@"]
Expand Down
Loading