From 21792520e727676e4d4e8bd24a8fe29da4dab152 Mon Sep 17 00:00:00 2001 From: Neil Schemenauer Date: Fri, 24 Apr 2026 10:24:05 -0700 Subject: [PATCH] [Build] Add Python 3.14 to supported version list. (#34770) Signed-off-by: Neil Schemenauer Co-authored-by: Simon Mo --- CMakeLists.txt | 2 +- docker/Dockerfile.nightly_torch | 2 +- pyproject.toml | 3 ++- requirements/common.txt | 2 +- requirements/cpu.txt | 2 +- requirements/cuda.txt | 2 +- requirements/rocm.txt | 4 ++-- requirements/test/cuda.in | 2 +- requirements/test/cuda.txt | 4 ++-- requirements/test/nightly-torch.txt | 2 +- requirements/test/rocm.in | 2 +- requirements/test/rocm.txt | 6 +++--- requirements/test/xpu.txt | 4 ++-- requirements/xpu.txt | 2 +- 14 files changed, 20 insertions(+), 19 deletions(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index e79c5b9f912..272c48f7ff9 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -34,7 +34,7 @@ install(CODE "set(CMAKE_INSTALL_LOCAL_ONLY TRUE)" ALL_COMPONENTS) # Supported python versions. These versions will be searched in order, the # first match will be selected. These should be kept in sync with setup.py. # -set(PYTHON_SUPPORTED_VERSIONS "3.10" "3.11" "3.12" "3.13") +set(PYTHON_SUPPORTED_VERSIONS "3.10" "3.11" "3.12" "3.13" "3.14") # Supported AMD GPU architectures. set(HIP_SUPPORTED_ARCHS "gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151;gfx1152;gfx1153;gfx1200;gfx1201") diff --git a/docker/Dockerfile.nightly_torch b/docker/Dockerfile.nightly_torch index fa23b236538..6b9d85c9d17 100644 --- a/docker/Dockerfile.nightly_torch +++ b/docker/Dockerfile.nightly_torch @@ -77,7 +77,7 @@ RUN --mount=type=cache,target=/root/.cache/uv \ uv pip install --system $pkgs --index-url https://download.pytorch.org/whl/nightly/cu128 RUN --mount=type=cache,target=/root/.cache/uv \ - uv pip install --system numba==0.61.2 + uv pip install --system numba==0.65.0 RUN --mount=type=cache,target=/root/.cache/uv \ uv pip install --system -r requirements/common.txt diff --git a/pyproject.toml b/pyproject.toml index c7b77f0e91d..b8d14463256 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -24,13 +24,14 @@ classifiers = [ "Programming Language :: Python :: 3.11", "Programming Language :: Python :: 3.12", "Programming Language :: Python :: 3.13", + "Programming Language :: Python :: 3.14", "Intended Audience :: Developers", "Intended Audience :: Information Technology", "Intended Audience :: Science/Research", "Topic :: Scientific/Engineering :: Artificial Intelligence", "Topic :: Scientific/Engineering :: Information Analysis", ] -requires-python = ">=3.10,<3.14" +requires-python = ">=3.10,<3.15" dynamic = [ "version", "dependencies", "optional-dependencies"] [project.urls] diff --git a/requirements/common.txt b/requirements/common.txt index 6e7fd90d023..5d4519204ee 100644 --- a/requirements/common.txt +++ b/requirements/common.txt @@ -20,7 +20,7 @@ prometheus-fastapi-instrumentator >= 7.0.0 tiktoken >= 0.6.0 # Required for DBRX tokenizer lm-format-enforcer == 0.11.3 llguidance >= 1.3.0, < 1.4.0; platform_machine == "x86_64" or platform_machine == "arm64" or platform_machine == "aarch64" or platform_machine == "ppc64le" -outlines_core == 0.2.11 +outlines_core == 0.2.14 # required for outlines backend disk cache diskcache == 5.6.3 lark == 1.2.2 diff --git a/requirements/cpu.txt b/requirements/cpu.txt index 26a23ba628e..5ec338af736 100644 --- a/requirements/cpu.txt +++ b/requirements/cpu.txt @@ -4,7 +4,7 @@ setuptools==77.0.3 # this version can reuse CMake build dir -numba == 0.61.2; platform_machine != "s390x" # Required for N-gram speculative decoding +numba == 0.65.0; platform_machine != "s390x" # Required for N-gram speculative decoding # Dependencies for CPUs torch==2.11.0+cpu; platform_machine == "x86_64" or platform_machine == "s390x" or platform_machine == "aarch64" diff --git a/requirements/cuda.txt b/requirements/cuda.txt index 02d5ccab87e..2145b023769 100644 --- a/requirements/cuda.txt +++ b/requirements/cuda.txt @@ -1,7 +1,7 @@ # Common dependencies -r common.txt -numba == 0.61.2 # Required for N-gram speculative decoding +numba == 0.65.0 # Required for N-gram speculative decoding # Dependencies for NVIDIA GPUs torch==2.11.0 diff --git a/requirements/rocm.txt b/requirements/rocm.txt index 6639e71a4b9..0b472b90c02 100644 --- a/requirements/rocm.txt +++ b/requirements/rocm.txt @@ -5,7 +5,7 @@ grpcio==1.78.0 grpcio-reflection==1.78.0 -numba == 0.61.2 # Required for N-gram speculative decoding +numba == 0.65.0 # Required for N-gram speculative decoding # Dependencies for AMD GPUs datasets @@ -20,4 +20,4 @@ conch-triton-kernels==1.2.1 timm>=1.0.17 # amd-quark: required for Quark quantization on ROCm # To be consistent with test_quark.py -amd-quark>=0.8.99 \ No newline at end of file +amd-quark>=0.8.99 diff --git a/requirements/test/cuda.in b/requirements/test/cuda.in index 30076ff7071..73d50104d86 100644 --- a/requirements/test/cuda.in +++ b/requirements/test/cuda.in @@ -54,7 +54,7 @@ grpcio==1.78.0 grpcio-reflection==1.78.0 arctic-inference == 0.1.1 # Required for suffix decoding test -numba == 0.61.2 # Required for N-gram speculative decoding +numba == 0.65.0 # Required for N-gram speculative decoding numpy runai-model-streamer[s3,gcs,azure]==0.15.7 fastsafetensors>=0.2.2 # 0.2.2 contains important fixes for multi-GPU mem usage diff --git a/requirements/test/cuda.txt b/requirements/test/cuda.txt index 449939b965c..ffb0d9d00c3 100644 --- a/requirements/test/cuda.txt +++ b/requirements/test/cuda.txt @@ -479,7 +479,7 @@ lightning-utilities==0.14.3 # lightning # pytorch-lightning # torchmetrics -llvmlite==0.44.0 +llvmlite==0.47.0 # via numba lm-eval==0.4.11 # via -r requirements/test/cuda.in @@ -550,7 +550,7 @@ nltk==3.9.1 # via rouge-score num2words==0.5.14 # via -r requirements/test/cuda.in -numba==0.61.2 +numba==0.65.0 # via # -c requirements/cuda.txt # -r requirements/test/cuda.in diff --git a/requirements/test/nightly-torch.txt b/requirements/test/nightly-torch.txt index 420fb496a71..0c34cf01203 100644 --- a/requirements/test/nightly-torch.txt +++ b/requirements/test/nightly-torch.txt @@ -40,7 +40,7 @@ buildkite-test-collector==0.1.9 genai_perf>=0.0.8 tritonclient>=2.51.0 -numba == 0.61.2 # Required for N-gram speculative decoding +numba == 0.65.0 # Required for N-gram speculative decoding numpy runai-model-streamer[s3,gcs,azure]==0.15.7 fastsafetensors>=0.2.2 diff --git a/requirements/test/rocm.in b/requirements/test/rocm.in index 0ae2e487853..b7329b5a9a4 100644 --- a/requirements/test/rocm.in +++ b/requirements/test/rocm.in @@ -52,7 +52,7 @@ grpcio==1.78.0 grpcio-reflection==1.78.0 arctic-inference==0.1.1 # Required for suffix decoding test -numba==0.61.2 # Required for N-gram speculative decoding +numba==0.65.0 # Required for N-gram speculative decoding numpy runai-model-streamer[s3,gcs,azure]==0.15.7 fastsafetensors @ git+https://github.com/foundation-model-stack/fastsafetensors.git@0.2.2 # PyPI only ships CUDA wheels diff --git a/requirements/test/rocm.txt b/requirements/test/rocm.txt index 6558c2f3811..ca33e2d09aa 100644 --- a/requirements/test/rocm.txt +++ b/requirements/test/rocm.txt @@ -559,7 +559,7 @@ llguidance==1.3.0 # via # -c requirements/common.txt # -r requirements/test/../common.txt -llvmlite==0.44.0 +llvmlite==0.47.0 # via numba lm-eval==0.4.11 # via -r requirements/test/rocm.in @@ -655,7 +655,7 @@ nltk==3.9.3 # via rouge-score num2words==0.5.14 # via -r requirements/test/rocm.in -numba==0.61.2 +numba==0.65.0 # via # -c requirements/rocm.txt # -r requirements/test/rocm.in @@ -811,7 +811,7 @@ orjson==3.11.7 # via # genai-perf # kaleido -outlines-core==0.2.11 +outlines-core==0.2.14 # via # -c requirements/common.txt # -r requirements/test/../common.txt diff --git a/requirements/test/xpu.txt b/requirements/test/xpu.txt index 547269d4a2c..601838f843f 100644 --- a/requirements/test/xpu.txt +++ b/requirements/test/xpu.txt @@ -244,7 +244,7 @@ lazy-loader==0.5 # scikit-image librosa==0.10.2.post1 # via -r requirements/test/xpu.in -llvmlite==0.44.0 +llvmlite==0.47.0 # via numba lm-eval==0.4.11 # via -r requirements/test/xpu.in @@ -304,7 +304,7 @@ nltk==3.9.4 # via rouge-score num2words==0.5.14 # via -r requirements/test/xpu.in -numba==0.61.2 +numba==0.65.0 # via # -c requirements/xpu.txt # librosa diff --git a/requirements/xpu.txt b/requirements/xpu.txt index 3be85dcb5f4..0aef5842b45 100644 --- a/requirements/xpu.txt +++ b/requirements/xpu.txt @@ -9,7 +9,7 @@ setuptools>=77.0.3,<81.0.0 wheel jinja2>=3.1.6 datasets # for benchmark scripts -numba == 0.61.2 # Required for N-gram speculative decoding +numba == 0.65.0 # Required for N-gram speculative decoding --extra-index-url=https://download.pytorch.org/whl/xpu torch==2.11.0+xpu torchaudio