A high-throughput and memory-efficient inference and serving engine for LLMs
Mirrored from https://github.com/tduncan-lum/Nvidia-Optimization-Stack---VLLM.git
Synced Nov 5, 07:15 UTC
Add structure to `requirements/` directory (#39024)
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com> Co-authored-by: Robert Shaw <114415538+robertgshaw2-redhat@users.noreply.github.com>
Harry Mellor committed
Apr 10, 2026 at 21:46 UTC
fc645f1acc0f3250210738a026f812db37ae97f6
40 files changed
+354
-331
.buildkite/ci_config.yaml
+2
-2
@@ -8,8 +8,8 @@ run_all_patterns:
8
- "CMakeLists.txt"
9
- "requirements/common.txt"
10
- "requirements/cuda.txt"
11
- - "requirements/build.txt"
12
- - "requirements/test.txt"
11
+ - "requirements/build/cuda.txt"
12
+ - "requirements/test/cuda.txt"
13
- "setup.py"
14
- "csrc/"
15
- "cmake/"
.buildkite/ci_config_intel.yaml
+2
-2
@@ -6,8 +6,8 @@ run_all_patterns:
6
- "CMakeLists.txt"
7
- "requirements/common.txt"
8
- "requirements/xpu.txt"
9
- - "requirements/build.txt"
10
- - "requirements/test.txt"
9
+ - "requirements/build/cuda.txt"
10
+ - "requirements/test/cuda.txt"
11
- "setup.py"
12
- "csrc/"
13
- "cmake/"
.buildkite/scripts/hardware_ci/run-hpu-test.sh
+1
-1
@@ -42,7 +42,7 @@ WORKDIR /workspace/vllm
42
ENV no_proxy=localhost,127.0.0.1
43
ENV PT_HPU_ENABLE_LAZY_COLLECTIVES=true
44
45
-RUN bash -c 'pip install -r <(sed "/^torch/d" requirements/build.txt)'
45
+RUN bash -c 'pip install -r <(sed "/^torch/d" requirements/build/cuda.txt)'
46
RUN VLLM_TARGET_DEVICE=empty pip install --no-build-isolation -e .
47
RUN pip install git+https://github.com/vllm-project/vllm-gaudi.git
48
.buildkite/test-amd.yaml
+1
-1
@@ -123,7 +123,7 @@ steps:
123
soft_fail: true
124
working_dir: "/vllm-workspace/tests"
125
source_file_dependencies:
126
- - requirements/nightly_torch_test.txt
126
+ - requirements/test/nightly-torch.txt
127
- vllm/platforms/rocm.py
128
commands:
129
- bash standalone_tests/pytorch_nightly_dependency.sh
.buildkite/test_areas/pytorch.yaml
+1
-1
@@ -64,6 +64,6 @@ steps:
64
device: h200_18gb
65
soft_fail: true
66
source_file_dependencies:
67
- - requirements/nightly_torch_test.txt
67
+ - requirements/test/nightly-torch.txt
68
commands:
69
- bash standalone_tests/pytorch_nightly_dependency.sh
.buildkite/test_areas/quantization.yaml
+1
-1
@@ -9,7 +9,7 @@ steps:
9
- vllm/model_executor/layers/quantization
10
- tests/quantization
11
commands:
12
- # temporary install here since we need nightly, will move to requirements/test.in
12
+ # temporary install here since we need nightly, will move to requirements/test/cuda.in
13
# after torchao 0.12 release, and pin a working version of torchao nightly here
14
15
# since torchao nightly is only compatible with torch nightly currently
.github/workflows/macos-smoke-test.yml
+1
-1
@@ -32,7 +32,7 @@ jobs:
32
33
- name: Install dependencies and build vLLM
34
run: |
35
- uv pip install -r requirements/cpu-build.txt --index-strategy unsafe-best-match
35
+ uv pip install -r requirements/build/cpu.txt --index-strategy unsafe-best-match
36
uv pip install -r requirements/cpu.txt --index-strategy unsafe-best-match
37
uv pip install -e . --no-build-isolation
38
env:
.github/workflows/scripts/build.sh
+1
-1
@@ -9,7 +9,7 @@ PATH=${cuda_home}/bin:$PATH
9
LD_LIBRARY_PATH=${cuda_home}/lib64:$LD_LIBRARY_PATH
10
11
# Install requirements
12
-$python_executable -m pip install -r requirements/build.txt -r requirements/cuda.txt
12
+$python_executable -m pip install -r requirements/build/cuda.txt -r requirements/cuda.txt
13
14
# Limit the number of parallel jobs to avoid OOM
15
export MAX_JOBS=1
.gitignore
+1
@@ -29,6 +29,7 @@ __pycache__/
29
# Distribution / packaging
30
.Python
31
build/
32
+!requirements/build/
33
cmake-build-*/
34
CMakeUserPresets.json
35
develop-eggs/
.pre-commit-config.yaml
+29
-7
@@ -39,15 +39,24 @@ repos:
39
rev: 0.11.1
40
hooks:
41
- id: pip-compile
42
- args: [requirements/test.in, -c, requirements/common.txt, -o, requirements/test.txt, --index-strategy, unsafe-best-match, --torch-backend, cu130, --python-platform, x86_64-manylinux_2_28, --python-version, "3.12"]
43
- files: ^requirements/test\.(in|txt)$
42
+ args: [
43
+ requirements/test/cuda.in,
44
+ -c, requirements/cuda.txt,
45
+ -o, requirements/test/cuda.txt,
46
+ --index-strategy, unsafe-best-match,
47
+ --torch-backend, cu130,
48
+ --python-platform, x86_64-manylinux_2_28,
49
+ --python-version, "3.12",
50
+ ]
51
+ files: ^requirements/(common|cuda|test/cuda)\.(in|txt)$
52
- id: pip-compile
53
alias: pip-compile-rocm
54
name: pip-compile-rocm
55
args: [
48
- requirements/rocm-test.in, -o, requirements/rocm-test.txt,
49
- --index-strategy, unsafe-best-match,
56
+ requirements/test/rocm.in,
57
-c, requirements/rocm.txt,
58
+ -o, requirements/test/rocm.txt,
59
+ --index-strategy, unsafe-best-match,
60
--python-platform, x86_64-manylinux_2_28,
61
--python-version, "3.12",
62
# Exclude torch and CUDA/NVIDIA packages
@@ -108,14 +117,27 @@ repos:
117
--no-emit-package, nvidia-nvshmem-cu13,
118
--no-emit-package, nvidia-nvtx-cu13,
119
]
111
- files: ^requirements/rocm-test\.(in|txt)$
120
+ files: ^requirements/(common|rocm|test/rocm)\.(in|txt)$
121
+ - id: pip-compile
122
+ alias: pip-compile-xpu
123
+ name: pip-compile-xpu
124
+ args: [
125
+ requirements/test/xpu.in,
126
+ -c, requirements/xpu.txt,
127
+ -o, requirements/test/xpu.txt,
128
+ --index-strategy, unsafe-best-match,
129
+ --torch-backend, xpu,
130
+ --python-platform, x86_64-manylinux_2_39,
131
+ --python-version, "3.12",
132
+ ]
133
+ files: ^requirements/(common|xpu|test/xpu)\.(in|txt)$
134
- repo: local
135
hooks:
136
- id: format-torch-nightly-test
115
- name: reformat nightly_torch_test.txt to be in sync with test.in
137
+ name: reformat test/nightly-torch.txt to be in sync with test/cuda.in
138
language: python
139
entry: python tools/pre_commit/generate_nightly_torch_test.py
118
- files: ^requirements/test\.(in|txt)$
140
+ files: ^requirements/test/cuda\.(in|txt)$
141
- id: mypy-local
142
name: Run mypy locally for lowest supported Python version
143
entry: python tools/pre_commit/mypy.py 0 "3.10"
AGENTS.md
+3
-3
@@ -72,11 +72,11 @@ uv pip install -e . --torch-backend=auto
72
73
```bash
74
# Install test dependencies.
75
-# requirements/test.txt is pinned to x86_64; on other platforms, use the
75
+# requirements/test/cuda.txt is pinned to x86_64; on other platforms, use the
76
# unpinned source file instead:
77
-uv pip install -r requirements/test.in # resolves for current platform
77
+uv pip install -r requirements/test/cuda.in # resolves for current platform
78
# Or on x86_64:
79
-uv pip install -r requirements/test.txt
79
+uv pip install -r requirements/test/cuda.txt
80
81
# Run a specific test file (use .venv/bin/python directly;
82
# `source activate` does not persist in non-interactive shells):
docker/Dockerfile
+14
-14
@@ -204,7 +204,7 @@ ARG PYTORCH_CUDA_INDEX_BASE_URL
204
ARG PYTORCH_NIGHTLY
205
206
# Install build dependencies
207
-COPY requirements/build.txt requirements/build.txt
207
+COPY requirements/build/cuda.txt requirements/build/cuda.txt
208
COPY use_existing_torch.py use_existing_torch.py
209
COPY --from=base /workspace/torch_lib_versions.txt torch_lib_versions.txt
210
@@ -219,13 +219,13 @@ RUN --mount=type=cache,target=/root/.cache/uv \
219
if [ "${PYTORCH_NIGHTLY}" = "1" ]; then \
220
echo "Installing build requirements without torch..." \
221
&& python3 use_existing_torch.py --prefix \
222
- && uv pip install --python /opt/venv/bin/python3 -r requirements/build.txt \
222
+ && uv pip install --python /opt/venv/bin/python3 -r requirements/build/cuda.txt \
223
&& echo "Installing torch nightly..." \
224
&& uv pip install --python /opt/venv/bin/python3 $(cat torch_lib_versions.txt | grep -i "^torch=" | xargs) --pre \
225
--index-url ${PYTORCH_CUDA_INDEX_BASE_URL}/nightly/cu$(echo $CUDA_VERSION | cut -d. -f1,2 | tr -d '.'); \
226
else \
227
echo "Installing build requirements..." \
228
- && uv pip install --python /opt/venv/bin/python3 -r requirements/build.txt \
228
+ && uv pip install --python /opt/venv/bin/python3 -r requirements/build/cuda.txt \
229
--extra-index-url ${PYTORCH_CUDA_INDEX_BASE_URL}/cu$(echo $CUDA_VERSION | cut -d. -f1,2 | tr -d '.'); \
230
fi
231
@@ -355,7 +355,7 @@ ARG PYTORCH_CUDA_INDEX_BASE_URL
355
ARG PYTORCH_NIGHTLY
356
357
# Install build dependencies
358
-COPY requirements/build.txt requirements/build.txt
358
+COPY requirements/build/cuda.txt requirements/build/cuda.txt
359
COPY use_existing_torch.py use_existing_torch.py
360
COPY --from=base /workspace/torch_lib_versions.txt torch_lib_versions.txt
361
@@ -370,13 +370,13 @@ RUN --mount=type=cache,target=/root/.cache/uv \
370
if [ "${PYTORCH_NIGHTLY}" = "1" ]; then \
371
echo "Installing build requirements without torch..." \
372
&& python3 use_existing_torch.py --prefix \
373
- && uv pip install --python /opt/venv/bin/python3 -r requirements/build.txt \
373
+ && uv pip install --python /opt/venv/bin/python3 -r requirements/build/cuda.txt \
374
&& echo "Installing torch nightly..." \
375
&& uv pip install --python /opt/venv/bin/python3 $(cat torch_lib_versions.txt | grep -i "^torch=" | xargs) --pre \
376
--index-url ${PYTORCH_CUDA_INDEX_BASE_URL}/nightly/cu$(echo $CUDA_VERSION | cut -d. -f1,2 | tr -d '.'); \
377
else \
378
echo "Installing build requirements..." \
379
- && uv pip install --python /opt/venv/bin/python3 -r requirements/build.txt \
379
+ && uv pip install --python /opt/venv/bin/python3 -r requirements/build/cuda.txt \
380
--extra-index-url ${PYTORCH_CUDA_INDEX_BASE_URL}/cu$(echo $CUDA_VERSION | cut -d. -f1,2 | tr -d '.'); \
381
fi
382
@@ -450,8 +450,8 @@ ARG PYTORCH_NIGHTLY
450
451
# Install development dependencies
452
COPY requirements/lint.txt requirements/lint.txt
453
-COPY requirements/test.in requirements/test.in
454
-COPY requirements/test.txt requirements/test.txt
453
+COPY requirements/test/cuda.in requirements/test/cuda.in
454
+COPY requirements/test/cuda.txt requirements/test/cuda.txt
455
COPY requirements/dev.txt requirements/dev.txt
456
COPY use_existing_torch.py use_existing_torch.py
457
COPY --from=base /workspace/torch_lib_versions.txt torch_lib_versions.txt
@@ -459,8 +459,8 @@ RUN --mount=type=cache,target=/root/.cache/uv \
459
if [ "${PYTORCH_NIGHTLY}" = "1" ]; then \
460
echo "Installing dev requirements plus torch nightly..." \
461
&& python3 use_existing_torch.py --prefix \
462
- && cat torch_lib_versions.txt >> requirements/test.in \
463
- && uv pip compile requirements/test.in -o requirements/test.txt --index-strategy unsafe-best-match \
462
+ && cat torch_lib_versions.txt >> requirements/test/cuda.in \
463
+ && uv pip compile requirements/test/cuda.in -o requirements/test/cuda.txt --index-strategy unsafe-best-match \
464
--extra-index-url ${PYTORCH_CUDA_INDEX_BASE_URL}/nightly/cu$(echo $CUDA_VERSION | cut -d. -f1,2 | tr -d '.') \
465
&& uv pip install --python /opt/venv/bin/python3 $(cat torch_lib_versions.txt | xargs) --pre \
466
-r requirements/dev.txt \
@@ -727,8 +727,8 @@ ARG PYTORCH_NIGHTLY
727
728
# Install development dependencies (for testing)
729
COPY requirements/lint.txt requirements/lint.txt
730
-COPY requirements/test.in requirements/test.in
731
-COPY requirements/test.txt requirements/test.txt
730
+COPY requirements/test/cuda.in requirements/test/cuda.in
731
+COPY requirements/test/cuda.txt requirements/test/cuda.txt
732
COPY requirements/dev.txt requirements/dev.txt
733
COPY use_existing_torch.py use_existing_torch.py
734
COPY --from=base /workspace/torch_lib_versions.txt torch_lib_versions.txt
@@ -738,8 +738,8 @@ RUN --mount=type=cache,target=/root/.cache/uv \
738
if [ "${PYTORCH_NIGHTLY}" = "1" ]; then \
739
echo "Installing dev requirements plus torch nightly..." \
740
&& python3 use_existing_torch.py --prefix \
741
- && cat torch_lib_versions.txt >> requirements/test.in \
742
- && uv pip compile requirements/test.in -o requirements/test.txt --index-strategy unsafe-best-match \
741
+ && cat torch_lib_versions.txt >> requirements/test/cuda.in \
742
+ && uv pip compile requirements/test/cuda.in -o requirements/test/cuda.txt --index-strategy unsafe-best-match \
743
--extra-index-url ${PYTORCH_CUDA_INDEX_BASE_URL}/nightly/cu$(echo $CUDA_VERSION | cut -d. -f1,2 | tr -d '.') \
744
&& uv pip install --system $(cat torch_lib_versions.txt | xargs) --pre \
745
-r requirements/dev.txt \
docker/Dockerfile.cpu
+12
-12
@@ -107,10 +107,10 @@ RUN if [ "$TARGETARCH" = "arm64" ] && [ "$VLLM_CPU_X86" != "0" ]; then \
107
fi
108
109
# Copy build requirements
110
-COPY requirements/cpu-build.txt requirements/build.txt
110
+COPY requirements/build/cpu.txt requirements/build/cpu.txt
111
112
RUN --mount=type=cache,target=/root/.cache/uv \
113
- uv pip install -r requirements/build.txt
113
+ uv pip install -r requirements/build/cpu.txt
114
115
COPY . .
116
@@ -127,26 +127,26 @@ FROM base AS vllm-test-deps
127
WORKDIR /vllm-workspace
128
129
# Copy test requirements
130
-COPY requirements/test.in requirements/cpu-test.in
130
+COPY requirements/test/cuda.in requirements/test/cpu.in
131
132
RUN \
133
- sed -i '/mamba_ssm/d' requirements/cpu-test.in && \
133
+ sed -i '/mamba_ssm/d' requirements/test/cpu.in && \
134
remove_packages_not_supported_on_aarch64() { \
135
case "$(uname -m)" in \
136
aarch64|arm64) \
137
- sed -i '/decord/d' requirements/cpu-test.in; \
138
- sed -i '/terratorch/d' requirements/cpu-test.in; \
137
+ sed -i '/decord/d' requirements/test/cpu.in; \
138
+ sed -i '/terratorch/d' requirements/test/cpu.in; \
139
;; \
140
esac; \
141
}; \
142
remove_packages_not_supported_on_aarch64 && \
143
- sed -i 's/^torch==.*/torch==2.11.0/g' requirements/cpu-test.in && \
144
- sed -i 's/torchaudio.*/torchaudio/g' requirements/cpu-test.in && \
145
- sed -i 's/torchvision.*/torchvision/g' requirements/cpu-test.in && \
146
- uv pip compile requirements/cpu-test.in -o requirements/cpu-test.txt --index-strategy unsafe-best-match --torch-backend cpu
143
+ sed -i 's/^torch==.*/torch==2.11.0/g' requirements/test/cpu.in && \
144
+ sed -i 's/torchaudio.*/torchaudio/g' requirements/test/cpu.in && \
145
+ sed -i 's/torchvision.*/torchvision/g' requirements/test/cpu.in && \
146
+ uv pip compile requirements/test/cpu.in -o requirements/test/cpu.txt --index-strategy unsafe-best-match --torch-backend cpu
147
148
RUN --mount=type=cache,target=/root/.cache/uv \
149
- uv pip install -r requirements/cpu-test.txt
149
+ uv pip install -r requirements/test/cpu.txt
150
151
######################### DEV IMAGE #########################
152
FROM vllm-build AS vllm-dev
@@ -168,7 +168,7 @@ RUN --mount=type=cache,target=/root/.cache/uv \
168
--mount=type=bind,source=.git,target=.git \
169
VLLM_TARGET_DEVICE=cpu python3 setup.py develop
170
171
-COPY --from=vllm-test-deps /vllm-workspace/requirements/cpu-test.txt requirements/test.txt
171
+COPY --from=vllm-test-deps /vllm-workspace/requirements/test/cpu.txt requirements/test/cpu.txt
172
173
RUN --mount=type=cache,target=/root/.cache/uv \
174
uv pip install -r requirements/dev.txt && \
docker/Dockerfile.nightly_torch
+3
-3
@@ -107,7 +107,7 @@ COPY . .
107
RUN python3 use_existing_torch.py
108
109
RUN --mount=type=cache,target=/root/.cache/uv \
110
- uv pip install --system -r requirements/build.txt
110
+ uv pip install --system -r requirements/build/cuda.txt
111
112
ARG GIT_REPO_CHECK=0
113
RUN --mount=type=bind,source=.git,target=.git \
@@ -261,7 +261,7 @@ FROM vllm-base as test
261
COPY tests/ tests/
262
263
# install build and runtime dependencies without stable torch version
264
-COPY requirements/nightly_torch_test.txt requirements/nightly_torch_test.txt
264
+COPY requirements/test/nightly-torch.txt requirements/test/nightly-torch.txt
265
266
# This timeout (in seconds) is necessary when installing some dependencies via uv since it's likely to time out
267
# Reference: https://github.com/astral-sh/uv/pull/1694
@@ -277,7 +277,7 @@ RUN --mount=type=cache,target=/root/.cache/uv \
277
ENV HF_HUB_ENABLE_HF_TRANSFER 1
278
279
RUN --mount=type=cache,target=/root/.cache/uv \
280
- uv pip install --system -r requirements/nightly_torch_test.txt
280
+ uv pip install --system -r requirements/test/nightly-torch.txt
281
282
# Logging to confirm the torch versions
283
RUN pip freeze | grep -E 'torch|vllm|flashinfer'
docker/Dockerfile.ppc64le
+1
-1
@@ -251,7 +251,7 @@ RUN --mount=type=cache,target=/root/.cache/uv \
251
make -C /numactl install && \
252
# sentencepiece.pc is in some pkgconfig inside uv cache
253
export PKG_CONFIG_PATH=$(find / -type d -name "pkgconfig" 2>/dev/null | tr '\n' ':') && \
254
- nanobind_DIR=$(uv pip show nanobind | grep Location | sed 's/^Location: //;s/$/\/nanobind\/cmake/') && uv pip install -r /src/requirements/common.txt -r /src/requirements/cpu.txt -r /src/requirements/build.txt --no-build-isolation && \
254
+ nanobind_DIR=$(uv pip show nanobind | grep Location | sed 's/^Location: //;s/$/\/nanobind\/cmake/') && uv pip install -r /src/requirements/common.txt -r /src/requirements/cpu.txt -r /src/requirements/build/cuda.txt --no-build-isolation && \
255
cd /src/ && \
256
uv build --wheel --out-dir /vllmwheel/ --no-build-isolation && \
257
uv pip install /vllmwheel/*.whl
docker/Dockerfile.rocm
+1
-1
@@ -329,7 +329,7 @@ RUN --mount=type=bind,from=export_vllm,src=/,target=/install \
329
--mount=type=cache,target=/root/.cache/uv \
330
cd /install \
331
&& uv pip install --system -r requirements/rocm.txt \
332
- && uv pip install --system -r requirements/rocm-test.txt \
332
+ && uv pip install --system -r requirements/test/rocm.txt \
333
&& pip uninstall -y vllm \
334
&& uv pip install --system *.whl
335
docker/Dockerfile.s390x
+1
-1
@@ -262,7 +262,7 @@ RUN --mount=type=cache,target=/root/.cache/uv \
262
$OPENCV_WHL_FILE \
263
$OUTLINES_CORE_WHL_FILE \
264
--index-strategy unsafe-best-match \
265
- -r requirements/cpu-build.txt \
265
+ -r requirements/build/cpu.txt \
266
-r requirements/cpu.txt
267
268
docker/Dockerfile.xpu
+2
-8
@@ -76,20 +76,14 @@ ENV UV_LINK_MODE="copy"
76
RUN --mount=type=cache,target=/root/.cache/uv \
77
--mount=type=bind,src=requirements/common.txt,target=/workspace/vllm/requirements/common.txt \
78
--mount=type=bind,src=requirements/xpu.txt,target=/workspace/vllm/requirements/xpu.txt \
79
- --mount=type=bind,src=requirements/xpu-test.in,target=/workspace/vllm/requirements/xpu-test.in \
79
+ --mount=type=bind,src=requirements/test/xpu.txt,target=/workspace/vllm/requirements/test/xpu.txt \
80
uv pip install --upgrade pip && \
81
uv pip install -r requirements/xpu.txt && \
82
- uv pip compile /workspace/vllm/requirements/xpu-test.in \
83
- -o /workspace/vllm/requirements/xpu-test.txt \
84
- -c /workspace/vllm/requirements/xpu.txt \
85
- --index-strategy unsafe-best-match \
86
- --extra-index-url ${PIP_EXTRA_INDEX_URL} \
87
- --python-version ${PYTHON_VERSION} && \
82
uv pip install grpcio-tools protobuf nanobind && \
83
source /opt/intel/oneapi/setvars.sh --force && \
84
source /opt/intel/oneapi/ccl/2021.15/env/vars.sh --force && \
85
export CMAKE_PREFIX_PATH="$(python3 -c 'import site; print(site.getsitepackages()[0])'):${CMAKE_PREFIX_PATH}" && \
92
- uv pip install --no-build-isolation -r /workspace/vllm/requirements/xpu-test.txt
86
+ uv pip install --no-build-isolation -r /workspace/vllm/requirements/test/xpu.txt
87
88
89
docs/contributing/README.md
+2
-2
@@ -49,10 +49,10 @@ If you are developing vLLM's Python and CUDA/C++ code, install Pytorch first:
49
uv pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu129
50
```
51
52
-Then install the necessary build dependencies from `requirements/build.txt`, skipping `torch` as it was installed in the previous step:
52
+Then install the necessary build dependencies from `requirements/build/cuda.txt`, skipping `torch` as it was installed in the previous step:
53
54
```bash
55
-grep -v '^torch==' requirements/build.txt | uv pip install -r -
55
+grep -v '^torch==' requirements/build/cuda.txt | uv pip install -r -
56
```
57
58
Finally install vLLM using:
docs/contributing/incremental_build.md
+2
-2
@@ -16,10 +16,10 @@ Before setting up the incremental build:
16
17
2. **CUDA Toolkit:** Verify that the NVIDIA CUDA Toolkit is correctly installed and `nvcc` is accessible in your `PATH`. CMake relies on `nvcc` to compile CUDA code. You can typically find `nvcc` in `$CUDA_HOME/bin/nvcc` or by running `which nvcc`. If you encounter issues, refer to the [official CUDA Toolkit installation guides](https://developer.nvidia.com/cuda-toolkit-archive) and vLLM's main [GPU installation documentation](../getting_started/installation/gpu.md#troubleshooting) for troubleshooting. The `CMAKE_CUDA_COMPILER` variable in your `CMakeUserPresets.json` should also point to your `nvcc` binary.
18
19
-3. **Build Tools:** It is highly recommended to install `ccache` for fast rebuilds by caching compilation results (e.g., `sudo apt install ccache` or `conda install ccache`). Also, ensure the core build dependencies like `cmake` and `ninja` are installed. These are installable through `requirements/build.txt` or your system's package manager.
19
+3. **Build Tools:** It is highly recommended to install `ccache` for fast rebuilds by caching compilation results (e.g., `sudo apt install ccache` or `conda install ccache`). Also, ensure the core build dependencies like `cmake` and `ninja` are installed. These are installable through `requirements/build/cuda.txt` or your system's package manager.
20
21
```console
22
- uv pip install -r requirements/build.txt --torch-backend=auto
22
+ uv pip install -r requirements/build/cuda.txt --torch-backend=auto
23
```
24
25
## Setting up the CMake Build Environment
docs/getting_started/installation/cpu.arm.inc.md
+2
-2
@@ -96,14 +96,14 @@ cd vllm_source
96
Third, install required dependencies:
97
98
```bash
99
-uv pip install -r requirements/cpu-build.txt --torch-backend cpu
99
+uv pip install -r requirements/build/cpu.txt --torch-backend cpu
100
uv pip install -r requirements/cpu.txt --torch-backend cpu
101
```
102
103
??? console "pip"
104
```bash
105
pip install --upgrade pip
106
- pip install -v -r requirements/cpu-build.txt --extra-index-url https://download.pytorch.org/whl/cpu
106
+ pip install -v -r requirements/build/cpu.txt --extra-index-url https://download.pytorch.org/whl/cpu
107
pip install -v -r requirements/cpu.txt --extra-index-url https://download.pytorch.org/whl/cpu
108
```
109
docs/getting_started/installation/cpu.s390x.inc.md
+4
-4
@@ -46,10 +46,10 @@ Execute the following commands to build and install vLLM from source.
46
Please build the following dependencies, `torchvision`, `pyarrow` from source before building vLLM.
47
48
```bash
49
- sed -i '/^torch/d' requirements/build.txt # remove torch from requirements/build.txt since we use nightly builds
49
+ sed -i '/^torch/d' requirements/build/cuda.txt # remove torch from requirements/build/cuda.txt since we use nightly builds
50
uv pip install -v \
51
--torch-backend auto \
52
- -r requirements/build.txt \
52
+ -r requirements/build/cuda.txt \
53
-r requirements/cpu.txt \
54
VLLM_TARGET_DEVICE=cpu python setup.py bdist_wheel && \
55
uv pip install dist/*.whl
@@ -57,10 +57,10 @@ Execute the following commands to build and install vLLM from source.
57
58
??? console "pip"
59
```bash
60
- sed -i '/^torch/d' requirements/build.txt # remove torch from requirements/build.txt since we use nightly builds
60
+ sed -i '/^torch/d' requirements/build/cuda.txt # remove torch from requirements/build/cuda.txt since we use nightly builds
61
pip install -v \
62
--extra-index-url https://download.pytorch.org/whl/nightly/cpu \
63
- -r requirements/build.txt \
63
+ -r requirements/build/cuda.txt \
64
-r requirements/cpu.txt \
65
VLLM_TARGET_DEVICE=cpu python setup.py bdist_wheel && \
66
pip install dist/*.whl
docs/getting_started/installation/cpu.x86.inc.md
+2
-2
@@ -88,14 +88,14 @@ cd vllm_source
88
Install the required dependencies:
89
90
```bash
91
-uv pip install -r requirements/cpu-build.txt --torch-backend cpu
91
+uv pip install -r requirements/build/cpu.txt --torch-backend cpu
92
uv pip install -r requirements/cpu.txt --torch-backend cpu
93
```
94
95
??? console "pip"
96
```bash
97
pip install --upgrade pip
98
- pip install -v -r requirements/cpu-build.txt --extra-index-url https://download.pytorch.org/whl/cpu
98
+ pip install -v -r requirements/build/cpu.txt --extra-index-url https://download.pytorch.org/whl/cpu
99
pip install -v -r requirements/cpu.txt --extra-index-url https://download.pytorch.org/whl/cpu
100
```
101
docs/getting_started/installation/gpu.cuda.inc.md
+1
-1
@@ -162,7 +162,7 @@ To build vLLM using an existing PyTorch installation:
162
git clone https://github.com/vllm-project/vllm.git
163
cd vllm
164
python use_existing_torch.py
165
-uv pip install -r requirements/build.txt
165
+uv pip install -r requirements/build/cuda.txt
166
uv pip install --no-build-isolation -e .
167
```
168
docs/mkdocs/hooks/generate_argparse.py
+1
-1
@@ -46,7 +46,7 @@ mock_if_no_torch(
46
47
48
# Mock any version checks by reading from compiled CI requirements
49
-with open(ROOT_DIR / "requirements/test.txt") as f:
49
+with open(ROOT_DIR / "requirements/test/cuda.txt") as f:
50
VERSIONS = dict(line.strip().split("==") for line in f if "==" in line)
51
importlib.metadata.version = lambda name: VERSIONS.get(name) or "0.0.0"
52
pyproject.toml
+1
-1
@@ -1,5 +1,5 @@
1
[build-system]
2
-# Should be mirrored in requirements/build.txt
2
+# Should be mirrored in requirements/build/cuda.txt
3
requires = [
4
"cmake>=3.26.1",
5
"ninja",
requirements/build/cpu.txt
renamed
requirements/build/cuda.txt
renamed
requirements/build/rocm.txt
renamed
+1
-1
@@ -1,5 +1,5 @@
1
# Common dependencies
2
--r common.txt
2
+-r ../common.txt
3
4
5
--extra-index-url https://download.pytorch.org/whl/rocm7.1
requirements/dev.txt
+1
-1
@@ -1,5 +1,5 @@
1
-r lint.txt
2
--r test.txt
2
+-r test/cuda.txt
3
4
# Avoid adding requirements directly to this file.
5
# Instead, modify the two files referenced above.
requirements/test/cuda.in
renamed
requirements/test/cuda.txt
renamed
+71
-65
@@ -1,5 +1,5 @@
1
# This file was autogenerated by uv via the following command:
2
-# uv pip compile requirements/test.in -c requirements/common.txt -o requirements/test.txt --index-strategy unsafe-best-match --torch-backend cu130 --python-platform x86_64-manylinux_2_28 --python-version 3.12
2
+# uv pip compile requirements/test/cuda.in -c requirements/cuda.txt -o requirements/test/cuda.txt --index-strategy unsafe-best-match --torch-backend cu130 --python-platform x86_64-manylinux_2_28 --python-version 3.12
3
absl-py==2.1.0
4
# via
5
# rouge-score
@@ -29,7 +29,7 @@ albucore==0.0.16
29
# via terratorch
30
albumentations==1.4.6
31
# via
32
- # -r requirements/test.in
32
+ # -r requirements/test/cuda.in
33
# terratorch
34
alembic==1.16.4
35
# via optuna
@@ -46,7 +46,7 @@ anyio==4.6.2.post1
46
# httpx
47
# starlette
48
arctic-inference==0.1.1
49
- # via -r requirements/test.in
49
+ # via -r requirements/test/cuda.in
50
argcomplete==3.5.1
51
# via datamodel-code-generator
52
arrow==1.3.0
@@ -64,7 +64,7 @@ attrs==24.2.0
64
audioread==3.0.1
65
# via librosa
66
av==16.1.0
67
- # via -r requirements/test.in
67
+ # via -r requirements/test/cuda.in
68
azure-core==1.38.2
69
# via
70
# azure-identity
@@ -75,16 +75,16 @@ azure-storage-blob==12.28.0
75
# via runai-model-streamer-azure
76
backoff==2.2.1
77
# via
78
- # -r requirements/test.in
78
+ # -r requirements/test/cuda.in
79
# schemathesis
80
bitsandbytes==0.49.2
81
# via
82
- # -r requirements/test.in
82
+ # -r requirements/test/cuda.in
83
# lightning
84
black==24.10.0
85
# via datamodel-code-generator
86
blobfile==3.0.0
87
- # via -r requirements/test.in
87
+ # via -r requirements/test/cuda.in
88
bm25s==0.2.13
89
# via mteb
90
boto3==1.35.57
@@ -98,7 +98,7 @@ botocore==1.35.57
98
bounded-pool-executor==0.0.3
99
# via pqdm
100
buildkite-test-collector==0.1.9
101
- # via -r requirements/test.in
101
+ # via -r requirements/test/cuda.in
102
cachetools==5.5.2
103
# via google-auth
104
certifi==2024.8.30
@@ -176,21 +176,21 @@ cupy-cuda12x==13.6.0
176
cycler==0.12.1
177
# via matplotlib
178
datamodel-code-generator==0.26.3
179
- # via -r requirements/test.in
179
+ # via -r requirements/test/cuda.in
180
dataproperty==1.0.1
181
# via
182
# pytablewriter
183
# tabledata
184
datasets==3.3.0
185
# via
186
- # -r requirements/test.in
186
+ # -r requirements/test/cuda.in
187
# evaluate
188
# lm-eval
189
# mteb
190
decorator==5.1.1
191
# via librosa
192
decord==0.6.0
193
- # via -r requirements/test.in
193
+ # via -r requirements/test/cuda.in
194
diffusers==0.36.0
195
# via terratorch
196
dill==0.3.8
@@ -211,7 +211,7 @@ docstring-parser==0.17.0
211
# via jsonargparse
212
einops==0.8.1
213
# via
214
- # -r requirements/test.in
214
+ # -r requirements/test/cuda.in
215
# encodec
216
# terratorch
217
# torchgeo
@@ -236,7 +236,9 @@ fastparquet==2024.11.0
236
fastrlock==0.8.2
237
# via cupy-cuda12x
238
fastsafetensors==0.2.2
239
- # via -r requirements/test.in
239
+ # via
240
+ # -c requirements/cuda.txt
241
+ # -r requirements/test/cuda.in
242
filelock==3.16.1
243
# via
244
# -c requirements/common.txt
@@ -273,7 +275,7 @@ fsspec==2024.12.0
275
ftfy==6.3.1
276
# via open-clip-torch
277
genai-perf==0.0.16
276
- # via -r requirements/test.in
278
+ # via -r requirements/test/cuda.in
279
genson==1.3.0
280
# via datamodel-code-generator
281
geopandas==1.0.1
@@ -306,19 +308,19 @@ google-resumable-media==2.7.2
308
googleapis-common-protos==1.70.0
309
# via google-api-core
310
gpt-oss==0.0.8
309
- # via -r requirements/test.in
311
+ # via -r requirements/test/cuda.in
312
graphql-core==3.2.6
313
# via hypothesis-graphql
314
greenlet==3.2.3
315
# via sqlalchemy
316
grpcio==1.78.0
317
# via
316
- # -r requirements/test.in
318
+ # -r requirements/test/cuda.in
319
# grpcio-reflection
320
# ray
321
# tensorboard
322
grpcio-reflection==1.78.0
321
- # via -r requirements/test.in
323
+ # via -r requirements/test/cuda.in
324
h11==0.14.0
325
# via
326
# httpcore
@@ -341,7 +343,7 @@ httpcore==1.0.6
343
# via httpx
344
httpx==0.27.2
345
# via
344
- # -r requirements/test.in
346
+ # -r requirements/test/cuda.in
347
# diffusers
348
# perceptron
349
# schemathesis
@@ -386,7 +388,7 @@ idna==3.10
388
# requests
389
# yarl
390
imagehash==4.3.2
389
- # via -r requirements/test.in
391
+ # via -r requirements/test/cuda.in
392
imageio==2.37.0
393
# via scikit-image
394
importlib-metadata==8.7.0
@@ -400,7 +402,7 @@ inflect==5.6.2
402
iniconfig==2.0.0
403
# via pytest
404
instanttensor==0.1.5
403
- # via -r requirements/test.in
405
+ # via -r requirements/test/cuda.in
406
isodate==0.7.2
407
# via azure-storage-blob
408
isoduration==20.11.0
@@ -414,7 +416,7 @@ jinja2==3.1.6
416
# lm-eval
417
# torch
418
jiwer==3.0.5
417
- # via -r requirements/test.in
419
+ # via -r requirements/test/cuda.in
420
jmespath==1.0.1
421
# via
422
# boto3
@@ -445,7 +447,7 @@ jsonschema-specifications==2024.10.1
447
junit-xml==1.9
448
# via schemathesis
449
kaldi-native-fbank==1.22.3
448
- # via -r requirements/test.in
450
+ # via -r requirements/test/cuda.in
451
kaleido==0.2.1
452
# via genai-perf
453
kiwisolver==1.4.7
@@ -461,7 +463,7 @@ lazy-loader==0.4
463
libnacl==2.1.0
464
# via tensorizer
465
librosa==0.10.2.post1
464
- # via -r requirements/test.in
466
+ # via -r requirements/test/cuda.in
467
lightly==1.5.22
468
# via
469
# terratorch
@@ -480,7 +482,7 @@ lightning-utilities==0.14.3
482
llvmlite==0.44.0
483
# via numba
484
lm-eval==0.4.11
483
- # via -r requirements/test.in
485
+ # via -r requirements/test/cuda.in
486
lxml==5.3.0
487
# via
488
# blobfile
@@ -499,7 +501,7 @@ markupsafe==3.0.1
501
# werkzeug
502
matplotlib==3.9.2
503
# via
502
- # -r requirements/test.in
504
+ # -r requirements/test/cuda.in
505
# lightning
506
# pycocotools
507
# torchgeo
@@ -513,7 +515,7 @@ mdurl==0.1.2
515
mistral-common==1.11.0
516
# via
517
# -c requirements/common.txt
516
- # -r requirements/test.in
518
+ # -r requirements/test/cuda.in
519
more-itertools==10.5.0
520
# via lm-eval
521
mpmath==1.3.0
@@ -529,7 +531,7 @@ msgpack==1.1.0
531
# librosa
532
# ray
533
mteb==2.8.3
532
- # via -r requirements/test.in
534
+ # via -r requirements/test/cuda.in
535
multidict==6.1.0
536
# via
537
# aiohttp
@@ -547,15 +549,16 @@ networkx==3.2.1
549
nltk==3.9.1
550
# via rouge-score
551
num2words==0.5.14
550
- # via -r requirements/test.in
552
+ # via -r requirements/test/cuda.in
553
numba==0.61.2
554
# via
553
- # -r requirements/test.in
555
+ # -c requirements/cuda.txt
556
+ # -r requirements/test/cuda.in
557
# librosa
558
# resampy
559
numpy==2.2.6
560
# via
558
- # -r requirements/test.in
561
+ # -r requirements/test/cuda.in
562
# accelerate
563
# albucore
564
# albumentations
@@ -661,7 +664,7 @@ omegaconf==2.3.0
664
# hydra-core
665
# lightning
666
open-clip-torch==2.32.0
664
- # via -r requirements/test.in
667
+ # via -r requirements/test/cuda.in
668
openai-harmony==0.0.4
669
# via
670
# -c requirements/common.txt
@@ -673,12 +676,12 @@ opencensus-context==0.1.3
676
opencv-python-headless==4.13.0.90
677
# via
678
# -c requirements/common.txt
676
- # -r requirements/test.in
679
+ # -r requirements/test/cuda.in
680
# albucore
681
# albumentations
682
# mistral-common
683
openpyxl==3.1.5
681
- # via -r requirements/test.in
684
+ # via -r requirements/test/cuda.in
685
opentelemetry-api==1.35.0
686
# via
687
# -c requirements/common.txt
@@ -754,9 +757,9 @@ pathvalidate==3.2.1
757
patsy==1.0.1
758
# via statsmodels
759
peft==0.16.0
757
- # via -r requirements/test.in
760
+ # via -r requirements/test/cuda.in
761
perceptron==0.1.4
759
- # via -r requirements/test.in
762
+ # via -r requirements/test/cuda.in
763
perf-analyzer==0.1.0
764
# via genai-perf
765
pillow==10.4.0
@@ -782,7 +785,7 @@ platformdirs==4.3.6
785
# wandb
786
plotly==5.24.1
787
# via
785
- # -r requirements/test.in
788
+ # -r requirements/test/cuda.in
789
# genai-perf
790
pluggy==1.5.0
791
# via
@@ -795,7 +798,7 @@ pooch==1.8.2
798
portalocker==2.10.1
799
# via sacrebleu
800
pqdm==0.2.0
798
- # via -r requirements/test.in
801
+ # via -r requirements/test/cuda.in
802
prometheus-client==0.22.0
803
# via
804
# -c requirements/common.txt
@@ -852,7 +855,7 @@ pycryptodomex==3.22.0
855
pydantic==2.12.0
856
# via
857
# -c requirements/common.txt
855
- # -r requirements/test.in
858
+ # -r requirements/test/cuda.in
859
# albumentations
860
# datamodel-code-generator
861
# fastapi
@@ -891,7 +894,7 @@ pytablewriter==1.2.0
894
# via lm-eval
895
pytest==8.3.5
896
# via
894
- # -r requirements/test.in
897
+ # -r requirements/test/cuda.in
898
# buildkite-test-collector
899
# genai-perf
900
# pytest-asyncio
@@ -904,21 +907,21 @@ pytest==8.3.5
907
# pytest-timeout
908
# schemathesis
909
pytest-asyncio==0.24.0
907
- # via -r requirements/test.in
910
+ # via -r requirements/test/cuda.in
911
pytest-cov==6.3.0
909
- # via -r requirements/test.in
912
+ # via -r requirements/test/cuda.in
913
pytest-forked==1.6.0
911
- # via -r requirements/test.in
914
+ # via -r requirements/test/cuda.in
915
pytest-mock==3.14.0
916
# via genai-perf
917
pytest-rerunfailures==14.0
915
- # via -r requirements/test.in
918
+ # via -r requirements/test/cuda.in
919
pytest-shard==0.1.2
917
- # via -r requirements/test.in
920
+ # via -r requirements/test/cuda.in
921
pytest-subtests==0.14.1
922
# via schemathesis
923
pytest-timeout==2.3.1
921
- # via -r requirements/test.in
924
+ # via -r requirements/test/cuda.in
925
python-box==7.3.2
926
# via terratorch
927
python-dateutil==2.9.0.post0
@@ -972,7 +975,7 @@ rasterio==1.4.3
975
# terratorch
976
# torchgeo
977
ray==2.48.0
975
- # via -r requirements/test.in
978
+ # via -r requirements/test/cuda.in
979
redis==5.2.0
980
# via tensorizer
981
referencing==0.35.1
@@ -1015,7 +1018,7 @@ requests==2.32.3
1018
# transformers
1019
# wandb
1020
resampy==0.4.3
1018
- # via -r requirements/test.in
1021
+ # via -r requirements/test/cuda.in
1022
responses==0.25.3
1023
# via genai-perf
1024
rfc3339-validator==0.1.4
@@ -1043,7 +1046,7 @@ rsa==4.9.1
1046
rtree==1.4.0
1047
# via torchgeo
1048
runai-model-streamer==0.15.7
1046
- # via -r requirements/test.in
1049
+ # via -r requirements/test/cuda.in
1050
runai-model-streamer-azure==0.15.7
1051
# via runai-model-streamer
1052
runai-model-streamer-gcs==0.15.7
@@ -1064,7 +1067,7 @@ safetensors==0.4.5
1067
# timm
1068
# transformers
1069
schemathesis==3.39.15
1067
- # via -r requirements/test.in
1070
+ # via -r requirements/test/cuda.in
1071
scikit-image==0.25.2
1072
# via
1073
# albumentations
@@ -1091,12 +1094,12 @@ scipy==1.13.1
1094
# vocos
1095
segmentation-models-pytorch==0.5.0
1096
# via
1094
- # -r requirements/test.in
1097
+ # -r requirements/test/cuda.in
1098
# terratorch
1099
# torchgeo
1100
sentence-transformers==5.2.0
1101
# via
1099
- # -r requirements/test.in
1102
+ # -r requirements/test/cuda.in
1103
# mteb
1104
sentry-sdk==2.52.0
1105
# via wandb
@@ -1136,7 +1139,7 @@ sortedcontainers==2.4.0
1139
# via hypothesis
1140
soundfile==0.12.1
1141
# via
1139
- # -r requirements/test.in
1142
+ # -r requirements/test/cuda.in
1143
# genai-perf
1144
# librosa
1145
# mistral-common
@@ -1172,7 +1175,7 @@ tabulate==0.9.0
1175
tacoreader==0.5.6
1176
# via terratorch
1177
tblib==3.1.0
1175
- # via -r requirements/test.in
1178
+ # via -r requirements/test/cuda.in
1179
tcolorpy==0.1.6
1180
# via pytablewriter
1181
tenacity==9.1.2
@@ -1187,13 +1190,13 @@ tensorboard-data-server==0.7.2
1190
tensorboardx==2.6.4
1191
# via lightning
1192
tensorizer==2.10.1
1190
- # via -r requirements/test.in
1193
+ # via -r requirements/test/cuda.in
1194
termcolor==3.1.0
1195
# via
1196
# gpt-oss
1197
# terratorch
1198
terratorch==1.2.2
1196
- # via -r requirements/test.in
1199
+ # via -r requirements/test/cuda.in
1200
threadpoolctl==3.5.0
1201
# via scikit-learn
1202
tifffile==2025.3.30
@@ -1208,7 +1211,7 @@ tiktoken==0.12.0
1211
# mistral-common
1212
timm==1.0.17
1213
# via
1211
- # -r requirements/test.in
1214
+ # -r requirements/test/cuda.in
1215
# open-clip-torch
1216
# segmentation-models-pytorch
1217
# terratorch
@@ -1216,7 +1219,7 @@ timm==1.0.17
1219
tokenizers==0.22.0
1220
# via
1221
# -c requirements/common.txt
1219
- # -r requirements/test.in
1222
+ # -r requirements/test/cuda.in
1223
# transformers
1224
tomli==2.2.1
1225
# via schemathesis
@@ -1224,7 +1227,8 @@ tomli-w==1.2.0
1227
# via schemathesis
1228
torch==2.11.0+cu130
1229
# via
1227
- # -r requirements/test.in
1230
+ # -c requirements/cuda.txt
1231
+ # -r requirements/test/cuda.in
1232
# accelerate
1233
# bitsandbytes
1234
# encodec
@@ -1249,7 +1253,8 @@ torch==2.11.0+cu130
1253
# vocos
1254
torchaudio==2.11.0+cu130
1255
# via
1252
- # -r requirements/test.in
1256
+ # -c requirements/cuda.txt
1257
+ # -r requirements/test/cuda.in
1258
# encodec
1259
# vocos
1260
torchgeo==0.7.0
@@ -1262,7 +1267,8 @@ torchmetrics==1.7.4
1267
# torchgeo
1268
torchvision==0.26.0+cu130
1269
# via
1265
- # -r requirements/test.in
1270
+ # -c requirements/cuda.txt
1271
+ # -r requirements/test/cuda.in
1272
# lightly
1273
# open-clip-torch
1274
# segmentation-models-pytorch
@@ -1292,17 +1298,17 @@ tqdm==4.67.3
1298
transformers==4.57.5
1299
# via
1300
# -c requirements/common.txt
1295
- # -r requirements/test.in
1301
+ # -r requirements/test/cuda.in
1302
# genai-perf
1303
# peft
1304
# sentence-transformers
1305
# transformers-stream-generator
1306
transformers-stream-generator==0.0.5
1301
- # via -r requirements/test.in
1307
+ # via -r requirements/test/cuda.in
1308
triton==3.6.0
1309
# via torch
1310
tritonclient==2.64.0
1305
- # via -r requirements/test.in
1311
+ # via -r requirements/test/cuda.in
1312
typepy==1.3.2
1313
# via
1314
# dataproperty
@@ -1371,11 +1377,11 @@ urllib3==2.2.3
1377
uvicorn==0.35.0
1378
# via gpt-oss
1379
vector-quantize-pytorch==1.21.2
1374
- # via -r requirements/test.in
1380
+ # via -r requirements/test/cuda.in
1381
virtualenv==20.31.2
1382
# via ray
1383
vocos==0.1.0
1378
- # via -r requirements/test.in
1384
+ # via -r requirements/test/cuda.in
1385
wandb==0.24.2
1386
# via terratorch
1387
wcwidth==0.2.13
requirements/test/nightly-torch.txt
renamed
requirements/test/rocm.in
renamed
+2
-2
@@ -1,4 +1,4 @@
1
--r common.txt
1
+-r ../common.txt
2
3
# testing
4
pytest
@@ -78,7 +78,7 @@ datasets>=3.3.0,<=3.6.0
78
openpyxl # required for perf comparison excel report
79
plotly # required for perf comparison html report
80
81
-# ROCm-specific extras (not in CUDA test.in)
81
+# ROCm-specific extras (not in CUDA cuda.in)
82
rapidfuzz
83
torchgeo==0.7.0
84
multiprocess==0.70.16
requirements/test/rocm.txt
renamed
+122
-122
@@ -1,5 +1,5 @@
1
# This file was autogenerated by uv via the following command:
2
-# uv pip compile requirements/rocm-test.in -o requirements/rocm-test.txt --index-strategy unsafe-best-match -c requirements/rocm.txt --python-platform x86_64-manylinux_2_28 --python-version 3.12 --no-emit-package torch --no-emit-package torchvision --no-emit-package torchaudio --no-emit-package triton --no-emit-package cuda-bindings --no-emit-package cuda-pathfinder --no-emit-package cuda-toolkit --no-emit-package cupy-cuda12x --no-emit-package nvidia-cublas --no-emit-package nvidia-cuda-cupti --no-emit-package nvidia-cuda-nvrtc --no-emit-package nvidia-cuda-runtime --no-emit-package nvidia-cudnn --no-emit-package nvidia-cufft --no-emit-package nvidia-cufile --no-emit-package nvidia-curand --no-emit-package nvidia-cusolver --no-emit-package nvidia-cusparse --no-emit-package nvidia-cusparselt --no-emit-package nvidia-nccl --no-emit-package nvidia-nvjitlink --no-emit-package nvidia-nvshmem --no-emit-package nvidia-nvtx --no-emit-package nvidia-cublas-cu12 --no-emit-package nvidia-cuda-cupti-cu12 --no-emit-package nvidia-cuda-nvrtc-cu12 --no-emit-package nvidia-cuda-runtime-cu12 --no-emit-package nvidia-cudnn-cu12 --no-emit-package nvidia-cufft-cu12 --no-emit-package nvidia-cufile-cu12 --no-emit-package nvidia-curand-cu12 --no-emit-package nvidia-cusolver-cu12 --no-emit-package nvidia-cusparse-cu12 --no-emit-package nvidia-cusparselt-cu12 --no-emit-package nvidia-nccl-cu12 --no-emit-package nvidia-nvjitlink-cu12 --no-emit-package nvidia-nvshmem-cu12 --no-emit-package nvidia-nvtx-cu12 --no-emit-package nvidia-cublas-cu13 --no-emit-package nvidia-cuda-cupti-cu13 --no-emit-package nvidia-cuda-nvrtc-cu13 --no-emit-package nvidia-cuda-runtime-cu13 --no-emit-package nvidia-cudnn-cu13 --no-emit-package nvidia-cufft-cu13 --no-emit-package nvidia-cufile-cu13 --no-emit-package nvidia-curand-cu13 --no-emit-package nvidia-cusolver-cu13 --no-emit-package nvidia-cusparse-cu13 --no-emit-package nvidia-cusparselt-cu13 --no-emit-package nvidia-nccl-cu13 --no-emit-package nvidia-nvjitlink-cu13 --no-emit-package nvidia-nvshmem-cu13 --no-emit-package nvidia-nvtx-cu13
2
+# uv pip compile requirements/test/rocm.in -c requirements/rocm.txt -o requirements/test/rocm.txt --index-strategy unsafe-best-match --python-platform x86_64-manylinux_2_28 --python-version 3.12 --no-emit-package torch --no-emit-package torchvision --no-emit-package torchaudio --no-emit-package triton --no-emit-package cuda-bindings --no-emit-package cuda-pathfinder --no-emit-package cuda-toolkit --no-emit-package cupy-cuda12x --no-emit-package nvidia-cublas --no-emit-package nvidia-cuda-cupti --no-emit-package nvidia-cuda-nvrtc --no-emit-package nvidia-cuda-runtime --no-emit-package nvidia-cudnn --no-emit-package nvidia-cufft --no-emit-package nvidia-cufile --no-emit-package nvidia-curand --no-emit-package nvidia-cusolver --no-emit-package nvidia-cusparse --no-emit-package nvidia-cusparselt --no-emit-package nvidia-nccl --no-emit-package nvidia-nvjitlink --no-emit-package nvidia-nvshmem --no-emit-package nvidia-nvtx --no-emit-package nvidia-cublas-cu12 --no-emit-package nvidia-cuda-cupti-cu12 --no-emit-package nvidia-cuda-nvrtc-cu12 --no-emit-package nvidia-cuda-runtime-cu12 --no-emit-package nvidia-cudnn-cu12 --no-emit-package nvidia-cufft-cu12 --no-emit-package nvidia-cufile-cu12 --no-emit-package nvidia-curand-cu12 --no-emit-package nvidia-cusolver-cu12 --no-emit-package nvidia-cusparse-cu12 --no-emit-package nvidia-cusparselt-cu12 --no-emit-package nvidia-nccl-cu12 --no-emit-package nvidia-nvjitlink-cu12 --no-emit-package nvidia-nvshmem-cu12 --no-emit-package nvidia-nvtx-cu12 --no-emit-package nvidia-cublas-cu13 --no-emit-package nvidia-cuda-cupti-cu13 --no-emit-package nvidia-cuda-nvrtc-cu13 --no-emit-package nvidia-cuda-runtime-cu13 --no-emit-package nvidia-cudnn-cu13 --no-emit-package nvidia-cufft-cu13 --no-emit-package nvidia-cufile-cu13 --no-emit-package nvidia-curand-cu13 --no-emit-package nvidia-cusolver-cu13 --no-emit-package nvidia-cusparse-cu13 --no-emit-package nvidia-cusparselt-cu13 --no-emit-package nvidia-nccl-cu13 --no-emit-package nvidia-nvjitlink-cu13 --no-emit-package nvidia-nvshmem-cu13 --no-emit-package nvidia-nvtx-cu13
3
absl-py==2.4.0
4
# via
5
# rouge-score
@@ -15,7 +15,7 @@ aiohappyeyeballs==2.6.1
15
aiohttp==3.13.3
16
# via
17
# -c requirements/common.txt
18
- # -r requirements/common.txt
18
+ # -r requirements/test/../common.txt
19
# aiohttp-cors
20
# fsspec
21
# gpt-oss
@@ -29,7 +29,7 @@ albucore==0.1.2
29
# via terratorch
30
albumentations==1.4.6
31
# via
32
- # -r requirements/rocm-test.in
32
+ # -r requirements/test/rocm.in
33
# terratorch
34
alembic==1.18.4
35
# via optuna
@@ -42,7 +42,7 @@ annotated-types==0.7.0
42
anthropic==0.89.0
43
# via
44
# -c requirements/common.txt
45
- # -r requirements/common.txt
45
+ # -r requirements/test/../common.txt
46
antlr4-python3-runtime==4.9.3
47
# via
48
# hydra-core
@@ -57,7 +57,7 @@ anyio==4.13.0
57
# starlette
58
# watchfiles
59
arctic-inference==0.1.1
60
- # via -r requirements/rocm-test.in
60
+ # via -r requirements/test/rocm.in
61
argcomplete==3.6.3
62
# via datamodel-code-generator
63
arrow==1.4.0
@@ -76,7 +76,7 @@ attrs==26.1.0
76
audioread==3.0.1
77
# via librosa
78
av==16.1.0
79
- # via -r requirements/rocm-test.in
79
+ # via -r requirements/test/rocm.in
80
azure-core==1.39.0
81
# via
82
# azure-identity
@@ -87,18 +87,18 @@ azure-storage-blob==12.28.0
87
# via runai-model-streamer-azure
88
backoff==2.2.1
89
# via
90
- # -r requirements/rocm-test.in
90
+ # -r requirements/test/rocm.in
91
# schemathesis
92
bitsandbytes==0.49.2
93
# via
94
- # -r requirements/rocm-test.in
94
+ # -r requirements/test/rocm.in
95
# lightning
96
black==26.3.1
97
# via datamodel-code-generator
98
blake3==1.0.8
99
- # via -r requirements/common.txt
99
+ # via -r requirements/test/../common.txt
100
blobfile==3.0.0
101
- # via -r requirements/rocm-test.in
101
+ # via -r requirements/test/rocm.in
102
bm25s==0.2.13
103
# via mteb
104
boto3==1.42.74
@@ -112,11 +112,11 @@ botocore==1.42.74
112
bounded-pool-executor==0.0.3
113
# via pqdm
114
buildkite-test-collector==0.1.9
115
- # via -r requirements/rocm-test.in
115
+ # via -r requirements/test/rocm.in
116
cachetools==7.0.5
117
- # via -r requirements/common.txt
117
+ # via -r requirements/test/../common.txt
118
cbor2==5.9.0
119
- # via -r requirements/common.txt
119
+ # via -r requirements/test/../common.txt
120
certifi==2026.2.25
121
# via
122
# fiona
@@ -162,7 +162,7 @@ cligj==0.7.2
162
# fiona
163
# rasterio
164
cloudpickle==3.1.2
165
- # via -r requirements/common.txt
165
+ # via -r requirements/test/../common.txt
166
colorama==0.4.6
167
# via
168
# perceptron
@@ -175,7 +175,7 @@ colorlog==6.10.1
175
compressed-tensors==0.14.0.1
176
# via
177
# -c requirements/common.txt
178
- # -r requirements/common.txt
178
+ # -r requirements/test/../common.txt
179
contourpy==1.3.3
180
# via matplotlib
181
coverage==7.13.5
@@ -192,25 +192,25 @@ cryptography==46.0.0
192
cycler==0.12.1
193
# via matplotlib
194
datamodel-code-generator==0.55.0
195
- # via -r requirements/rocm-test.in
195
+ # via -r requirements/test/rocm.in
196
dataproperty==1.1.0
197
# via
198
# pytablewriter
199
# tabledata
200
datasets==3.6.0
201
# via
202
- # -r requirements/rocm-test.in
202
+ # -r requirements/test/rocm.in
203
# evaluate
204
# lm-eval
205
# mteb
206
decorator==5.2.1
207
# via librosa
208
decord==0.6.0
209
- # via -r requirements/rocm-test.in
209
+ # via -r requirements/test/rocm.in
210
depyf==0.20.0
211
# via
212
# -c requirements/common.txt
213
- # -r requirements/common.txt
213
+ # -r requirements/test/../common.txt
214
diffusers==0.37.0
215
# via terratorch
216
dill==0.3.8
@@ -223,7 +223,7 @@ dill==0.3.8
223
diskcache==5.6.3
224
# via
225
# -c requirements/common.txt
226
- # -r requirements/common.txt
226
+ # -r requirements/test/../common.txt
227
distlib==0.4.0
228
# via virtualenv
229
distro==1.9.0
@@ -242,8 +242,8 @@ docstring-parser==0.17.0
242
# jsonargparse
243
einops==0.8.2
244
# via
245
- # -r requirements/common.txt
246
- # -r requirements/rocm-test.in
245
+ # -r requirements/test/../common.txt
246
+ # -r requirements/test/rocm.in
247
# encodec
248
# terratorch
249
# torchgeo
@@ -264,7 +264,7 @@ evaluate==0.4.6
264
fastapi==0.135.2
265
# via
266
# -c requirements/common.txt
267
- # -r requirements/common.txt
267
+ # -r requirements/test/../common.txt
268
# gpt-oss
269
# model-hosting-container-standards
270
fastapi-cli==0.0.24
@@ -278,11 +278,11 @@ fastparquet==2026.3.0
278
fastsafetensors==0.2.2
279
# via
280
# -c requirements/rocm.txt
281
- # -r requirements/rocm-test.in
281
+ # -r requirements/test/rocm.in
282
filelock==3.25.2
283
# via
284
# -c requirements/common.txt
285
- # -r requirements/common.txt
285
+ # -r requirements/test/../common.txt
286
# blobfile
287
# datasets
288
# diffusers
@@ -317,7 +317,7 @@ fsspec==2025.3.0
317
ftfy==6.3.1
318
# via open-clip-torch
319
genai-perf==0.0.16
320
- # via -r requirements/rocm-test.in
320
+ # via -r requirements/test/rocm.in
321
genson==1.3.0
322
# via datamodel-code-generator
323
geopandas==1.1.3
@@ -325,7 +325,7 @@ geopandas==1.1.3
325
gguf==0.18.0
326
# via
327
# -c requirements/common.txt
328
- # -r requirements/common.txt
328
+ # -r requirements/test/../common.txt
329
gitdb==4.0.12
330
# via gitpython
331
gitpython==3.1.46
@@ -357,7 +357,7 @@ googleapis-common-protos==1.73.0
357
# opentelemetry-exporter-otlp-proto-grpc
358
# opentelemetry-exporter-otlp-proto-http
359
gpt-oss==0.0.8
360
- # via -r requirements/rocm-test.in
360
+ # via -r requirements/test/rocm.in
361
graphql-core==3.2.8
362
# via hypothesis-graphql
363
greenlet==3.3.2
@@ -365,7 +365,7 @@ greenlet==3.3.2
365
grpcio==1.78.0
366
# via
367
# -c requirements/rocm.txt
368
- # -r requirements/rocm-test.in
368
+ # -r requirements/test/rocm.in
369
# grpcio-reflection
370
# opentelemetry-exporter-otlp-proto-grpc
371
# ray
@@ -373,7 +373,7 @@ grpcio==1.78.0
373
grpcio-reflection==1.78.0
374
# via
375
# -c requirements/rocm.txt
376
- # -r requirements/rocm-test.in
376
+ # -r requirements/test/rocm.in
377
h11==0.16.0
378
# via
379
# httpcore
@@ -398,7 +398,7 @@ httptools==0.7.1
398
# via uvicorn
399
httpx==0.27.2
400
# via
401
- # -r requirements/rocm-test.in
401
+ # -r requirements/test/rocm.in
402
# anthropic
403
# diffusers
404
# fastapi
@@ -412,7 +412,7 @@ httpx-sse==0.4.3
412
# via mcp
413
huggingface-hub==0.36.2
414
# via
415
- # -r requirements/rocm-test.in
415
+ # -r requirements/test/rocm.in
416
# accelerate
417
# datasets
418
# diffusers
@@ -452,9 +452,9 @@ idna==3.11
452
# requests
453
# yarl
454
ijson==3.5.0
455
- # via -r requirements/common.txt
455
+ # via -r requirements/test/../common.txt
456
imagehash==4.3.2
457
- # via -r requirements/rocm-test.in
457
+ # via -r requirements/test/rocm.in
458
imageio==2.37.3
459
# via scikit-image
460
importlib-metadata==8.7.1
@@ -468,7 +468,7 @@ inflect==7.5.0
468
iniconfig==2.3.0
469
# via pytest
470
instanttensor==0.1.6
471
- # via -r requirements/rocm-test.in
471
+ # via -r requirements/test/rocm.in
472
interegular==0.3.3
473
# via lm-format-enforcer
474
isodate==0.7.2
@@ -489,7 +489,7 @@ jiter==0.13.0
489
# anthropic
490
# openai
491
jiwer==4.0.0
492
- # via -r requirements/rocm-test.in
492
+ # via -r requirements/test/rocm.in
493
jmespath==1.1.0
494
# via
495
# boto3
@@ -522,7 +522,7 @@ jsonschema-specifications==2025.9.1
522
junit-xml==1.9
523
# via schemathesis
524
kaldi-native-fbank==1.22.3
525
- # via -r requirements/rocm-test.in
525
+ # via -r requirements/test/rocm.in
526
kaleido==1.0.0
527
# via genai-perf
528
kiwisolver==1.5.0
@@ -534,7 +534,7 @@ kornia-rs==0.1.10
534
lark==1.2.2
535
# via
536
# -c requirements/common.txt
537
- # -r requirements/common.txt
537
+ # -r requirements/test/../common.txt
538
lazy-loader==0.4
539
# via
540
# librosa
@@ -542,7 +542,7 @@ lazy-loader==0.4
542
libnacl==2.1.0
543
# via tensorizer
544
librosa==0.10.2.post1
545
- # via -r requirements/rocm-test.in
545
+ # via -r requirements/test/rocm.in
546
lightly==1.5.22
547
# via
548
# terratorch
@@ -561,15 +561,15 @@ lightning-utilities==0.15.3
561
llguidance==1.3.0
562
# via
563
# -c requirements/common.txt
564
- # -r requirements/common.txt
564
+ # -r requirements/test/../common.txt
565
llvmlite==0.44.0
566
# via numba
567
lm-eval==0.4.11
568
- # via -r requirements/rocm-test.in
568
+ # via -r requirements/test/rocm.in
569
lm-format-enforcer==0.11.3
570
# via
571
# -c requirements/common.txt
572
- # -r requirements/common.txt
572
+ # -r requirements/test/../common.txt
573
logistro==2.0.1
574
# via
575
# choreographer
@@ -594,7 +594,7 @@ markupsafe==3.0.3
594
# werkzeug
595
matplotlib==3.10.8
596
# via
597
- # -r requirements/rocm-test.in
597
+ # -r requirements/test/rocm.in
598
# lightning
599
# torchgeo
600
mbstrdecoder==1.1.4
@@ -603,18 +603,18 @@ mbstrdecoder==1.1.4
603
# pytablewriter
604
# typepy
605
mcp==1.27.0
606
- # via -r requirements/common.txt
606
+ # via -r requirements/test/../common.txt
607
mdurl==0.1.2
608
# via markdown-it-py
609
mistral-common==1.11.0
610
# via
611
# -c requirements/common.txt
612
- # -r requirements/common.txt
613
- # -r requirements/rocm-test.in
612
+ # -r requirements/test/../common.txt
613
+ # -r requirements/test/rocm.in
614
model-hosting-container-standards==0.1.14
615
# via
616
# -c requirements/common.txt
617
- # -r requirements/common.txt
617
+ # -r requirements/test/../common.txt
618
more-itertools==10.8.0
619
# via
620
# inflect
@@ -632,16 +632,16 @@ msgpack==1.1.2
632
# librosa
633
# ray
634
msgspec==0.20.0
635
- # via -r requirements/common.txt
635
+ # via -r requirements/test/../common.txt
636
mteb==2.11.5
637
- # via -r requirements/rocm-test.in
637
+ # via -r requirements/test/rocm.in
638
multidict==6.7.1
639
# via
640
# aiohttp
641
# yarl
642
multiprocess==0.70.16
643
# via
644
- # -r requirements/rocm-test.in
644
+ # -r requirements/test/rocm.in
645
# datasets
646
# evaluate
647
mypy-extensions==1.1.0
@@ -653,23 +653,23 @@ networkx==3.6.1
653
# scikit-image
654
# torch
655
ninja==1.13.0
656
- # via -r requirements/common.txt
656
+ # via -r requirements/test/../common.txt
657
nltk==3.9.3
658
# via rouge-score
659
num2words==0.5.14
660
- # via -r requirements/rocm-test.in
660
+ # via -r requirements/test/rocm.in
661
numba==0.61.2
662
# via
663
# -c requirements/rocm.txt
664
- # -r requirements/rocm-test.in
664
+ # -r requirements/test/rocm.in
665
# librosa
666
# resampy
667
numkong==7.1.1
668
# via albucore
669
numpy==2.2.6
670
# via
671
- # -r requirements/common.txt
672
- # -r requirements/rocm-test.in
671
+ # -r requirements/test/../common.txt
672
+ # -r requirements/test/rocm.in
673
# accelerate
674
# albucore
675
# albumentations
@@ -741,15 +741,15 @@ omegaconf==2.3.0
741
# hydra-core
742
# lightning
743
open-clip-torch==2.32.0
744
- # via -r requirements/rocm-test.in
744
+ # via -r requirements/test/rocm.in
745
openai==2.30.0
746
# via
747
# -c requirements/common.txt
748
- # -r requirements/common.txt
748
+ # -r requirements/test/../common.txt
749
openai-harmony==0.0.8
750
# via
751
# -c requirements/common.txt
752
- # -r requirements/common.txt
752
+ # -r requirements/test/../common.txt
753
# gpt-oss
754
opencensus==0.11.4
755
# via ray
@@ -758,16 +758,16 @@ opencensus-context==0.1.3
758
opencv-python-headless==4.13.0.92
759
# via
760
# -c requirements/common.txt
761
- # -r requirements/common.txt
762
- # -r requirements/rocm-test.in
761
+ # -r requirements/test/../common.txt
762
+ # -r requirements/test/rocm.in
763
# albumentations
764
# mistral-common
765
openpyxl==3.1.5
766
- # via -r requirements/rocm-test.in
766
+ # via -r requirements/test/rocm.in
767
opentelemetry-api==1.40.0
768
# via
769
# -c requirements/common.txt
770
- # -r requirements/common.txt
770
+ # -r requirements/test/../common.txt
771
# opentelemetry-exporter-otlp-proto-grpc
772
# opentelemetry-exporter-otlp-proto-http
773
# opentelemetry-exporter-prometheus
@@ -776,7 +776,7 @@ opentelemetry-api==1.40.0
776
opentelemetry-exporter-otlp==1.40.0
777
# via
778
# -c requirements/common.txt
779
- # -r requirements/common.txt
779
+ # -r requirements/test/../common.txt
780
opentelemetry-exporter-otlp-proto-common==1.40.0
781
# via
782
# opentelemetry-exporter-otlp-proto-grpc
@@ -796,7 +796,7 @@ opentelemetry-proto==1.40.0
796
opentelemetry-sdk==1.40.0
797
# via
798
# -c requirements/common.txt
799
- # -r requirements/common.txt
799
+ # -r requirements/test/../common.txt
800
# opentelemetry-exporter-otlp-proto-grpc
801
# opentelemetry-exporter-otlp-proto-http
802
# opentelemetry-exporter-prometheus
@@ -809,7 +809,7 @@ opentelemetry-semantic-conventions==0.61b0
809
opentelemetry-semantic-conventions-ai==0.5.1
810
# via
811
# -c requirements/common.txt
812
- # -r requirements/common.txt
812
+ # -r requirements/test/../common.txt
813
optuna==3.6.1
814
# via genai-perf
815
orjson==3.11.7
@@ -819,7 +819,7 @@ orjson==3.11.7
819
outlines-core==0.2.11
820
# via
821
# -c requirements/common.txt
822
- # -r requirements/common.txt
822
+ # -r requirements/test/../common.txt
823
packaging==26.0
824
# via
825
# -c requirements/rocm.txt
@@ -870,7 +870,7 @@ pandas==3.0.1
870
# torchgeo
871
# xarray
872
partial-json-parser==0.2.1.1.post7
873
- # via -r requirements/common.txt
873
+ # via -r requirements/test/../common.txt
874
pathspec==1.0.4
875
# via black
876
pathvalidate==3.3.1
@@ -878,14 +878,14 @@ pathvalidate==3.3.1
878
patsy==1.0.2
879
# via statsmodels
880
peft==0.18.1
881
- # via -r requirements/rocm-test.in
881
+ # via -r requirements/test/rocm.in
882
perceptron==0.1.4
883
- # via -r requirements/rocm-test.in
883
+ # via -r requirements/test/rocm.in
884
perf-analyzer==0.1.0
885
# via genai-perf
886
pillow==12.1.1
887
# via
888
- # -r requirements/common.txt
888
+ # -r requirements/test/../common.txt
889
# diffusers
890
# genai-perf
891
# imagehash
@@ -908,7 +908,7 @@ platformdirs==4.3.6
908
# wandb
909
plotly==6.6.0
910
# via
911
- # -r requirements/rocm-test.in
911
+ # -r requirements/test/rocm.in
912
# genai-perf
913
pluggy==1.6.0
914
# via
@@ -923,18 +923,18 @@ pooch==1.8.2
923
portalocker==3.2.0
924
# via sacrebleu
925
pqdm==0.2.0
926
- # via -r requirements/rocm-test.in
926
+ # via -r requirements/test/rocm.in
927
prometheus-client==0.24.1
928
# via
929
# -c requirements/common.txt
930
- # -r requirements/common.txt
930
+ # -r requirements/test/../common.txt
931
# opentelemetry-exporter-prometheus
932
# prometheus-fastapi-instrumentator
933
# ray
934
prometheus-fastapi-instrumentator==7.1.0
935
# via
936
# -c requirements/common.txt
937
- # -r requirements/common.txt
937
+ # -r requirements/test/../common.txt
938
propcache==0.4.1
939
# via
940
# aiohttp
@@ -944,7 +944,7 @@ proto-plus==1.27.1
944
protobuf==6.33.6
945
# via
946
# -c requirements/common.txt
947
- # -r requirements/common.txt
947
+ # -r requirements/test/../common.txt
948
# google-api-core
949
# googleapis-common-protos
950
# grpcio-reflection
@@ -957,14 +957,14 @@ protobuf==6.33.6
957
# wandb
958
psutil==7.2.2
959
# via
960
- # -r requirements/common.txt
960
+ # -r requirements/test/../common.txt
961
# accelerate
962
# peft
963
# tensorizer
964
py==1.11.0
965
# via pytest-forked
966
py-cpuinfo==9.0.0
967
- # via -r requirements/common.txt
967
+ # via -r requirements/test/../common.txt
968
py-spy==0.4.1
969
# via ray
970
pyarrow==23.0.1
@@ -978,7 +978,7 @@ pyasn1==0.6.3
978
pyasn1-modules==0.4.2
979
# via google-auth
980
pybase64==1.4.3
981
- # via -r requirements/common.txt
981
+ # via -r requirements/test/../common.txt
982
pycocotools==2.0.11
983
# via terratorch
984
pycountry==26.2.16
@@ -990,8 +990,8 @@ pycryptodomex==3.23.0
990
pydantic==2.12.5
991
# via
992
# -c requirements/common.txt
993
- # -r requirements/common.txt
994
- # -r requirements/rocm-test.in
993
+ # -r requirements/test/../common.txt
994
+ # -r requirements/test/rocm.in
995
# albumentations
996
# anthropic
997
# compressed-tensors
@@ -1047,7 +1047,7 @@ pytablewriter==1.2.1
1047
# via lm-eval
1048
pytest==8.3.5
1049
# via
1050
- # -r requirements/rocm-test.in
1050
+ # -r requirements/test/rocm.in
1051
# buildkite-test-collector
1052
# genai-perf
1053
# pytest-asyncio
@@ -1060,21 +1060,21 @@ pytest==8.3.5
1060
# pytest-timeout
1061
# schemathesis
1062
pytest-asyncio==0.24.0
1063
- # via -r requirements/rocm-test.in
1063
+ # via -r requirements/test/rocm.in
1064
pytest-cov==6.3.0
1065
- # via -r requirements/rocm-test.in
1065
+ # via -r requirements/test/rocm.in
1066
pytest-forked==1.6.0
1067
- # via -r requirements/rocm-test.in
1067
+ # via -r requirements/test/rocm.in
1068
pytest-mock==3.15.1
1069
# via genai-perf
1070
pytest-rerunfailures==14.0
1071
- # via -r requirements/rocm-test.in
1071
+ # via -r requirements/test/rocm.in
1072
pytest-shard==0.1.2
1073
- # via -r requirements/rocm-test.in
1073
+ # via -r requirements/test/rocm.in
1074
pytest-subtests==0.14.2
1075
# via schemathesis
1076
pytest-timeout==2.3.1
1077
- # via -r requirements/rocm-test.in
1077
+ # via -r requirements/test/rocm.in
1078
python-box==7.4.1
1079
# via terratorch
1080
python-dateutil==2.9.0.post0
@@ -1092,7 +1092,7 @@ python-dotenv==1.2.2
1092
# pydantic-settings
1093
# uvicorn
1094
python-json-logger==4.1.0
1095
- # via -r requirements/common.txt
1095
+ # via -r requirements/test/../common.txt
1096
python-multipart==0.0.22
1097
# via
1098
# fastapi
@@ -1113,7 +1113,7 @@ pywavelets==1.9.0
1113
# via imagehash
1114
pyyaml==6.0.3
1115
# via
1116
- # -r requirements/common.txt
1116
+ # -r requirements/test/../common.txt
1117
# accelerate
1118
# albumentations
1119
# datamodel-code-generator
@@ -1139,10 +1139,10 @@ pyyaml==6.0.3
1139
pyzmq==27.1.0
1140
# via
1141
# -c requirements/common.txt
1142
- # -r requirements/common.txt
1142
+ # -r requirements/test/../common.txt
1143
rapidfuzz==3.12.1
1144
# via
1145
- # -r requirements/rocm-test.in
1145
+ # -r requirements/test/rocm.in
1146
# jiwer
1147
rasterio==1.5.0
1148
# via
@@ -1150,7 +1150,7 @@ rasterio==1.5.0
1150
# terratorch
1151
# torchgeo
1152
ray==2.54.0
1153
- # via -r requirements/rocm-test.in
1153
+ # via -r requirements/test/rocm.in
1154
redis==7.3.0
1155
# via tensorizer
1156
referencing==0.37.0
@@ -1159,7 +1159,7 @@ referencing==0.37.0
1159
# jsonschema-specifications
1160
regex==2026.2.28
1161
# via
1162
- # -r requirements/common.txt
1162
+ # -r requirements/test/../common.txt
1163
# diffusers
1164
# nltk
1165
# open-clip-torch
@@ -1169,7 +1169,7 @@ regex==2026.2.28
1169
requests==2.32.5
1170
# via
1171
# -c requirements/common.txt
1172
- # -r requirements/common.txt
1172
+ # -r requirements/test/../common.txt
1173
# azure-core
1174
# buildkite-test-collector
1175
# datasets
@@ -1197,7 +1197,7 @@ requests==2.32.5
1197
# transformers
1198
# wandb
1199
resampy==0.4.3
1200
- # via -r requirements/rocm-test.in
1200
+ # via -r requirements/test/rocm.in
1201
responses==0.26.0
1202
# via genai-perf
1203
rfc3339-validator==0.1.4
@@ -1232,7 +1232,7 @@ rtree==1.4.1
1232
runai-model-streamer==0.15.7
1233
# via
1234
# -c requirements/rocm.txt
1235
- # -r requirements/rocm-test.in
1235
+ # -r requirements/test/rocm.in
1236
runai-model-streamer-azure==0.15.7
1237
# via runai-model-streamer
1238
runai-model-streamer-gcs==0.15.7
@@ -1253,7 +1253,7 @@ safetensors==0.7.0
1253
# timm
1254
# transformers
1255
schemathesis==3.39.15
1256
- # via -r requirements/rocm-test.in
1256
+ # via -r requirements/test/rocm.in
1257
scikit-image==0.26.0
1258
# via
1259
# albumentations
@@ -1281,26 +1281,26 @@ scipy==1.17.1
1281
# vocos
1282
segmentation-models-pytorch==0.5.0
1283
# via
1284
- # -r requirements/rocm-test.in
1284
+ # -r requirements/test/rocm.in
1285
# terratorch
1286
# torchgeo
1287
sentence-transformers==5.3.0
1288
# via
1289
- # -r requirements/rocm-test.in
1289
+ # -r requirements/test/rocm.in
1290
# mteb
1291
sentencepiece==0.2.1
1292
- # via -r requirements/common.txt
1292
+ # via -r requirements/test/../common.txt
1293
sentry-sdk==2.55.0
1294
# via
1295
# fastapi-cloud-cli
1296
# wandb
1297
setproctitle==1.3.7
1298
- # via -r requirements/common.txt
1298
+ # via -r requirements/test/../common.txt
1299
setuptools==79.0.1
1300
# via
1301
# -c requirements/common.txt
1302
# -c requirements/rocm.txt
1303
- # -r requirements/common.txt
1303
+ # -r requirements/test/../common.txt
1304
# model-hosting-container-standards
1305
# pytablewriter
1306
# tensorboard
@@ -1318,7 +1318,7 @@ simplejson==3.20.2
1318
six==1.17.0
1319
# via
1320
# -c requirements/common.txt
1321
- # -r requirements/common.txt
1321
+ # -r requirements/test/../common.txt
1322
# junit-xml
1323
# lightly
1324
# opencensus
@@ -1338,7 +1338,7 @@ sortedcontainers==2.4.0
1338
# via hypothesis
1339
soundfile==0.13.1
1340
# via
1341
- # -r requirements/rocm-test.in
1341
+ # -r requirements/test/rocm.in
1342
# genai-perf
1343
# librosa
1344
# mistral-common
@@ -1384,7 +1384,7 @@ tabulate==0.10.0
1384
tacoreader==0.5.6
1385
# via terratorch
1386
tblib==3.1.0
1387
- # via -r requirements/rocm-test.in
1387
+ # via -r requirements/test/rocm.in
1388
tcolorpy==0.1.7
1389
# via pytablewriter
1390
tenacity==9.1.4
@@ -1400,13 +1400,13 @@ tensorboardx==2.6.4
1400
tensorizer==2.10.1
1401
# via
1402
# -c requirements/rocm.txt
1403
- # -r requirements/rocm-test.in
1403
+ # -r requirements/test/rocm.in
1404
termcolor==3.3.0
1405
# via
1406
# gpt-oss
1407
# terratorch
1408
terratorch==1.2.2
1409
- # via -r requirements/rocm-test.in
1409
+ # via -r requirements/test/rocm.in
1410
threadpoolctl==3.6.0
1411
# via scikit-learn
1412
tifffile==2026.3.3
@@ -1416,14 +1416,14 @@ tifffile==2026.3.3
1416
tiktoken==0.12.0
1417
# via
1418
# -c requirements/common.txt
1419
- # -r requirements/common.txt
1419
+ # -r requirements/test/../common.txt
1420
# gpt-oss
1421
# lm-eval
1422
# mistral-common
1423
timm==1.0.17
1424
# via
1425
# -c requirements/rocm.txt
1426
- # -r requirements/rocm-test.in
1426
+ # -r requirements/test/rocm.in
1427
# open-clip-torch
1428
# segmentation-models-pytorch
1429
# terratorch
@@ -1431,8 +1431,8 @@ timm==1.0.17
1431
tokenizers==0.22.0
1432
# via
1433
# -c requirements/common.txt
1434
- # -r requirements/common.txt
1435
- # -r requirements/rocm-test.in
1434
+ # -r requirements/test/../common.txt
1435
+ # -r requirements/test/rocm.in
1436
# transformers
1437
tomli==2.4.0
1438
# via schemathesis
@@ -1440,7 +1440,7 @@ tomli-w==1.2.0
1440
# via schemathesis
1441
torchgeo==0.7.0
1442
# via
1443
- # -r requirements/rocm-test.in
1443
+ # -r requirements/test/rocm.in
1444
# terratorch
1445
torchmetrics==1.9.0
1446
# via
@@ -1450,7 +1450,7 @@ torchmetrics==1.9.0
1450
# torchgeo
1451
tqdm==4.67.3
1452
# via
1453
- # -r requirements/common.txt
1453
+ # -r requirements/test/../common.txt
1454
# datasets
1455
# evaluate
1456
# gguf
@@ -1474,8 +1474,8 @@ tqdm==4.67.3
1474
transformers==4.57.5
1475
# via
1476
# -c requirements/common.txt
1477
- # -r requirements/common.txt
1478
- # -r requirements/rocm-test.in
1477
+ # -r requirements/test/../common.txt
1478
+ # -r requirements/test/rocm.in
1479
# compressed-tensors
1480
# genai-perf
1481
# peft
@@ -1483,9 +1483,9 @@ transformers==4.57.5
1483
# transformers-stream-generator
1484
# xgrammar
1485
transformers-stream-generator==0.0.5
1486
- # via -r requirements/rocm-test.in
1486
+ # via -r requirements/test/rocm.in
1487
tritonclient==2.66.0
1488
- # via -r requirements/rocm-test.in
1488
+ # via -r requirements/test/rocm.in
1489
typeguard==4.5.1
1490
# via inflect
1491
typepy==1.3.4
@@ -1504,7 +1504,7 @@ typeshed-client==2.9.0
1504
typing-extensions==4.15.0
1505
# via
1506
# -c requirements/common.txt
1507
- # -r requirements/common.txt
1507
+ # -r requirements/test/../common.txt
1508
# aiosignal
1509
# albumentations
1510
# alembic
@@ -1577,16 +1577,16 @@ uvicorn==0.42.0
1577
uvloop==0.22.1
1578
# via uvicorn
1579
vector-quantize-pytorch==1.28.0
1580
- # via -r requirements/rocm-test.in
1580
+ # via -r requirements/test/rocm.in
1581
virtualenv==21.2.0
1582
# via ray
1583
vocos==0.1.0
1584
- # via -r requirements/rocm-test.in
1584
+ # via -r requirements/test/rocm.in
1585
wandb==0.25.1
1586
# via terratorch
1587
watchfiles==1.1.1
1588
# via
1589
- # -r requirements/common.txt
1589
+ # -r requirements/test/../common.txt
1590
# uvicorn
1591
wcwidth==0.6.0
1592
# via ftfy
@@ -1607,7 +1607,7 @@ xarray==2026.2.0
1607
xgrammar==0.1.33
1608
# via
1609
# -c requirements/common.txt
1610
- # -r requirements/common.txt
1610
+ # -r requirements/test/../common.txt
1611
xxhash==3.6.0
1612
# via
1613
# datasets
requirements/test/xpu.in
renamed
requirements/test/xpu.txt
renamed
+57
-57
@@ -1,11 +1,11 @@
1
# This file was autogenerated by uv via the following command:
2
-# uv pip compile requirements/xpu-test.in -o requirements/xpu-test.txt -c requirements/xpu.txt --python-version 3.12 --index-strategy unsafe-best-match
2
+# uv pip compile requirements/test/xpu.in -c requirements/xpu.txt -o requirements/test/xpu.txt --index-strategy unsafe-best-match --torch-backend xpu --python-platform x86_64-manylinux_2_39 --python-version 3.12
3
absl-py==2.4.0
4
# via
5
- # -r requirements/xpu-test.in
5
+ # -r requirements/test/xpu.in
6
# rouge-score
7
accelerate==1.13.0
8
- # via -r requirements/xpu-test.in
8
+ # via -r requirements/test/xpu.in
9
aiohappyeyeballs==2.6.1
10
# via aiohttp
11
aiohttp==3.13.4
@@ -17,7 +17,7 @@ aiohttp==3.13.4
17
aiosignal==1.4.0
18
# via aiohttp
19
albumentations==1.4.6
20
- # via -r requirements/xpu-test.in
20
+ # via -r requirements/test/xpu.in
21
annotated-doc==0.0.4
22
# via fastapi
23
annotated-types==0.7.0
@@ -27,7 +27,7 @@ anyio==4.13.0
27
# httpx
28
# starlette
29
arctic-inference==0.1.1
30
- # via -r requirements/xpu-test.in
30
+ # via -r requirements/test/xpu.in
31
attrs==26.1.0
32
# via
33
# aiohttp
@@ -36,13 +36,13 @@ attrs==26.1.0
36
# referencing
37
audioread==3.0.1
38
# via
39
- # -r requirements/xpu-test.in
39
+ # -r requirements/test/xpu.in
40
# librosa
41
blobfile==3.0.0
42
- # via -r requirements/xpu-test.in
42
+ # via -r requirements/test/xpu.in
43
bm25s==0.2.13
44
# via
45
- # -r requirements/xpu-test.in
45
+ # -r requirements/test/xpu.in
46
# mteb
47
bounded-pool-executor==0.0.3
48
# via pqdm
@@ -90,7 +90,7 @@ docker==7.1.0
90
# via gpt-oss
91
docopt==0.6.2
92
# via num2words
93
-dpcpp-cpp-rt==2025.3.1
93
+dpcpp-cpp-rt==2025.3.2
94
# via
95
# onemkl-sycl-blas
96
# onemkl-sycl-dft
@@ -124,7 +124,7 @@ fsspec==2026.2.0
124
# huggingface-hub
125
# torch
126
gpt-oss==0.0.8
127
- # via -r requirements/xpu-test.in
127
+ # via -r requirements/test/xpu.in
128
graphql-core==3.2.8
129
# via hypothesis-graphql
130
h11==0.16.0
@@ -134,7 +134,7 @@ h11==0.16.0
134
harfile==0.4.0
135
# via schemathesis
136
hf-transfer==0.1.9
137
- # via -r requirements/xpu-test.in
137
+ # via -r requirements/test/xpu.in
138
hf-xet==1.4.2
139
# via huggingface-hub
140
html2text==2025.4.15
@@ -171,27 +171,27 @@ idna==3.11
171
# yarl
172
imageio==2.37.3
173
# via scikit-image
174
-impi-rt==2021.17.0
174
+impi-rt==2021.17.2
175
# via
176
# oneccl
177
# torch
178
iniconfig==2.3.0
179
# via pytest
180
-intel-cmplr-lib-rt==2025.3.1
180
+intel-cmplr-lib-rt==2025.3.2
181
# via
182
# intel-sycl-rt
183
# torch
184
-intel-cmplr-lib-ur==2025.3.1
184
+intel-cmplr-lib-ur==2025.3.2
185
# via
186
# intel-openmp
187
# intel-sycl-rt
188
# torch
189
-intel-cmplr-lic-rt==2025.3.1
189
+intel-cmplr-lic-rt==2025.3.2
190
# via
191
# intel-opencl-rt
192
# intel-sycl-rt
193
# torch
194
-intel-opencl-rt==2025.3.1
194
+intel-opencl-rt==2025.3.2
195
# via
196
# dpcpp-cpp-rt
197
# onemkl-sycl-blas
@@ -200,14 +200,14 @@ intel-opencl-rt==2025.3.1
200
# onemkl-sycl-rng
201
# onemkl-sycl-sparse
202
# torch
203
-intel-openmp==2025.3.1
203
+intel-openmp==2025.3.2
204
# via
205
# dpcpp-cpp-rt
206
# mkl
207
# torch
208
-intel-pti==0.15.0
208
+intel-pti==0.16.0
209
# via torch
210
-intel-sycl-rt==2025.3.1
210
+intel-sycl-rt==2025.3.2
211
# via
212
# dpcpp-cpp-rt
213
# oneccl
@@ -218,7 +218,7 @@ jinja2==3.1.6
218
# lm-eval
219
# torch
220
jiwer==4.0.0
221
- # via -r requirements/xpu-test.in
221
+ # via -r requirements/test/xpu.in
222
joblib==1.5.3
223
# via
224
# librosa
@@ -242,11 +242,11 @@ lazy-loader==0.5
242
# librosa
243
# scikit-image
244
librosa==0.10.2.post1
245
- # via -r requirements/xpu-test.in
245
+ # via -r requirements/test/xpu.in
246
llvmlite==0.44.0
247
# via numba
248
lm-eval==0.4.11
249
- # via -r requirements/xpu-test.in
249
+ # via -r requirements/test/xpu.in
250
lxml==6.0.2
251
# via
252
# blobfile
@@ -265,11 +265,11 @@ mbstrdecoder==1.1.4
265
# typepy
266
mdurl==0.1.2
267
# via markdown-it-py
268
-mistral-common==1.10.0
268
+mistral-common==1.11.0
269
# via
270
# -c requirements/common.txt
271
- # -r requirements/xpu-test.in
272
-mkl==2025.3.0
271
+ # -r requirements/test/xpu.in
272
+mkl==2025.3.1
273
# via
274
# onemkl-sycl-blas
275
# onemkl-sycl-dft
@@ -278,7 +278,7 @@ mkl==2025.3.0
278
# onemkl-sycl-sparse
279
# torch
280
modelscope==1.35.3
281
- # via -r requirements/xpu-test.in
281
+ # via -r requirements/test/xpu.in
282
more-itertools==10.8.0
283
# via lm-eval
284
mpmath==1.3.0
@@ -286,7 +286,7 @@ mpmath==1.3.0
286
msgpack==1.1.2
287
# via librosa
288
mteb==2.12.7
289
- # via -r requirements/xpu-test.in
289
+ # via -r requirements/test/xpu.in
290
multidict==6.7.1
291
# via
292
# aiohttp
@@ -302,7 +302,7 @@ networkx==3.6.1
302
nltk==3.9.4
303
# via rouge-score
304
num2words==0.5.14
305
- # via -r requirements/xpu-test.in
305
+ # via -r requirements/test/xpu.in
306
numba==0.61.2
307
# via
308
# -c requirements/xpu.txt
@@ -334,28 +334,28 @@ numpy==2.2.6
334
# tifffile
335
# torchvision
336
# transformers
337
-oneccl==2021.17.1
337
+oneccl==2021.17.2
338
# via
339
# oneccl-devel
340
# torch
341
-oneccl-devel==2021.17.1
341
+oneccl-devel==2021.17.2
342
# via torch
343
-onemkl-license==2025.3.0
343
+onemkl-license==2025.3.1
344
# via
345
# mkl
346
# torch
347
-onemkl-sycl-blas==2025.3.0
347
+onemkl-sycl-blas==2025.3.1
348
# via
349
# onemkl-sycl-lapack
350
# onemkl-sycl-sparse
351
# torch
352
-onemkl-sycl-dft==2025.3.0
352
+onemkl-sycl-dft==2025.3.1
353
# via torch
354
-onemkl-sycl-lapack==2025.3.0
354
+onemkl-sycl-lapack==2025.3.1
355
# via torch
356
-onemkl-sycl-rng==2025.3.0
356
+onemkl-sycl-rng==2025.3.1
357
# via torch
358
-onemkl-sycl-sparse==2025.3.0
358
+onemkl-sycl-sparse==2025.3.1
359
# via torch
360
openai-harmony==0.0.8
361
# via
@@ -405,12 +405,12 @@ polars-runtime-32==1.39.3
405
# via polars
406
pooch==1.8.2
407
# via
408
- # -r requirements/xpu-test.in
408
+ # -r requirements/test/xpu.in
409
# librosa
410
portalocker==3.2.0
411
# via sacrebleu
412
pqdm==0.2.0
413
- # via -r requirements/xpu-test.in
413
+ # via -r requirements/test/xpu.in
414
propcache==0.4.1
415
# via
416
# aiohttp
@@ -451,13 +451,13 @@ pyrate-limiter==4.1.0
451
# via schemathesis
452
pystemmer==3.0.0
453
# via
454
- # -r requirements/xpu-test.in
454
+ # -r requirements/test/xpu.in
455
# mteb
456
pytablewriter==1.2.1
457
# via lm-eval
458
pytest==9.0.2
459
# via
460
- # -r requirements/xpu-test.in
460
+ # -r requirements/test/xpu.in
461
# pytest-asyncio
462
# pytest-cov
463
# pytest-forked
@@ -466,17 +466,17 @@ pytest==9.0.2
466
# pytest-timeout
467
# schemathesis
468
pytest-asyncio==1.3.0
469
- # via -r requirements/xpu-test.in
469
+ # via -r requirements/test/xpu.in
470
pytest-cov==6.3.0
471
- # via -r requirements/xpu-test.in
471
+ # via -r requirements/test/xpu.in
472
pytest-forked==1.6.0
473
- # via -r requirements/xpu-test.in
473
+ # via -r requirements/test/xpu.in
474
pytest-rerunfailures==14.0
475
- # via -r requirements/xpu-test.in
475
+ # via -r requirements/test/xpu.in
476
pytest-shard==0.1.2
477
- # via -r requirements/xpu-test.in
477
+ # via -r requirements/test/xpu.in
478
pytest-timeout==2.3.1
479
- # via -r requirements/xpu-test.in
479
+ # via -r requirements/test/xpu.in
480
python-dateutil==2.9.0.post0
481
# via
482
# pandas
@@ -496,7 +496,7 @@ pyyaml==6.0.3
496
# transformers
497
rapidfuzz==3.12.1
498
# via
499
- # -r requirements/xpu-test.in
499
+ # -r requirements/test/xpu.in
500
# jiwer
501
referencing==0.37.0
502
# via
@@ -543,7 +543,7 @@ safetensors==0.7.0
543
# timm
544
# transformers
545
schemathesis==4.14.2
546
- # via -r requirements/xpu-test.in
546
+ # via -r requirements/test/xpu.in
547
scikit-image==0.26.0
548
# via albumentations
549
scikit-learn==1.8.0
@@ -582,12 +582,12 @@ sortedcontainers==2.4.0
582
# via hypothesis
583
soundfile==0.13.1
584
# via
585
- # -r requirements/xpu-test.in
585
+ # -r requirements/test/xpu.in
586
# librosa
587
# mistral-common
588
soxr==0.5.0.post1
589
# via
590
- # -r requirements/xpu-test.in
590
+ # -r requirements/test/xpu.in
591
# librosa
592
# mistral-common
593
sqlitedict==2.1.0
@@ -606,13 +606,13 @@ tabledata==1.3.4
606
# via pytablewriter
607
tabulate==0.10.0
608
# via sacrebleu
609
-tbb==2022.3.0
609
+tbb==2022.3.1
610
# via
611
# intel-opencl-rt
612
# mkl
613
# torch
614
tblib==3.1.0
615
- # via -r requirements/xpu-test.in
615
+ # via -r requirements/test/xpu.in
616
tcmlib==1.4.1
617
# via
618
# tbb
@@ -638,12 +638,12 @@ tiktoken==0.12.0
638
# lm-eval
639
# mistral-common
640
timm==1.0.17
641
- # via -r requirements/xpu-test.in
641
+ # via -r requirements/test/xpu.in
642
tokenizers==0.22.2
643
# via
644
# -c requirements/common.txt
645
# transformers
646
-torch==2.10.0+xpu
646
+torch==2.11.0+xpu
647
# via
648
# -c requirements/xpu.txt
649
# accelerate
@@ -651,7 +651,7 @@ torch==2.10.0+xpu
651
# sentence-transformers
652
# timm
653
# torchvision
654
-torchvision==0.25.0+xpu
654
+torchvision==0.26.0+xpu
655
# via timm
656
tqdm==4.67.3
657
# via
@@ -669,7 +669,7 @@ transformers==4.57.6
669
# via
670
# -c requirements/common.txt
671
# sentence-transformers
672
-triton-xpu==3.6.0
672
+triton-xpu==3.7.0
673
# via torch
674
typepy==1.3.4
675
# via
@@ -704,7 +704,7 @@ typing-inspection==0.4.2
704
# via
705
# fastapi
706
# pydantic
707
-umf==1.0.2
707
+umf==1.0.3
708
# via
709
# intel-cmplr-lib-ur
710
# torch
tests/standalone_tests/pytorch_nightly_dependency.sh
+3
-3
@@ -28,8 +28,8 @@ uv pip freeze | grep -E '^torch|^torchvision|^torchaudio' | sort > before.txt
28
echo "Before:"
29
cat before.txt
30
31
-echo ">>> Installing requirements/nightly_torch_test.txt"
32
-uv pip install --quiet -r requirements/nightly_torch_test.txt
31
+echo ">>> Installing requirements/test/nightly-torch.txt"
32
+uv pip install --quiet -r requirements/test/nightly-torch.txt
33
34
echo ">>> Capturing torch-related versions after requirements install"
35
uv pip freeze | grep -E '^torch|^torchvision|^torchaudio' | sort > after.txt
@@ -40,7 +40,7 @@ echo ">>> Comparing versions"
40
if diff before.txt after.txt; then
41
echo "torch version not overridden."
42
else
43
- echo "torch version overridden by nightly_torch_test.txt, \
43
+ echo "torch version overridden by test/nightly-torch.txt, \
44
if the dependency is not triggered by the pytorch nightly test,\
45
please add the dependency to the list 'white_list' in tools/pre_commit/generate_nightly_torch_test.py"
46
exit 1
tools/generate_cmake_presets.py
+1
-1
@@ -128,7 +128,7 @@ def generate_presets(output_path="CMakeUserPresets.json", force_overwrite=False)
128
129
presets = {
130
"version": 6,
131
- # Keep in sync with CMakeLists.txt and requirements/build.txt
131
+ # Keep in sync with CMakeLists.txt and requirements/build/cuda.txt
132
"cmakeMinimumRequired": {"major": 3, "minor": 26, "patch": 1},
133
"configurePresets": [configure_preset],
134
"buildPresets": [
tools/pre_commit/generate_nightly_torch_test.py
+4
-4
@@ -3,15 +3,15 @@
3
"""
4
Generates specialized requirements files for nightly PyTorch testing.
5
6
-This script reads the main test requirements input file (`requirements/test.in`)
6
+This script reads the main test requirements input file (`requirements/test/cuda.in`)
7
and splits its content into two files:
8
-1. `requirements/nightly_torch_test.txt`: Contains dependencies
8
+1. `requirements/test/nightly-torch.txt`: Contains dependencies
9
except PyTorch-related.
10
2. `torch_nightly_test.txt`: Contains only PyTorch-related packages.
11
"""
12
13
-input_file = "requirements/test.in"
14
-output_file = "requirements/nightly_torch_test.txt"
13
+input_file = "requirements/test/cuda.in"
14
+output_file = "requirements/test/nightly-torch.txt"
15
16
# white list of packages that are not compatible with PyTorch nightly directly
17
# with pip install. Please add your package to this list if it is not compatible