diff --git a/docker/Dockerfile.rocm b/docker/Dockerfile.rocm index 545765ea856..16284e999a7 100644 --- a/docker/Dockerfile.rocm +++ b/docker/Dockerfile.rocm @@ -230,6 +230,7 @@ COPY --from=build_vllm ${COMMON_WORKDIR}/vllm/requirements /requirements COPY --from=build_vllm ${COMMON_WORKDIR}/vllm/benchmarks /benchmarks COPY --from=build_vllm ${COMMON_WORKDIR}/vllm/tests /tests COPY --from=build_vllm ${COMMON_WORKDIR}/vllm/examples /examples +COPY --from=build_vllm ${COMMON_WORKDIR}/vllm/tools/install_torchcodec_rocm.sh /tools/install_torchcodec_rocm.sh COPY --from=build_vllm ${COMMON_WORKDIR}/vllm/docker/Dockerfile.rocm /docker/ COPY --from=build_vllm ${COMMON_WORKDIR}/vllm/.buildkite /.buildkite COPY --from=build_vllm ${COMMON_WORKDIR}/vllm/pyproject.toml /pyproject.toml @@ -513,6 +514,7 @@ COPY --from=build_vllm_wheel_release ${COMMON_WORKDIR}/vllm/requirements /requir COPY --from=build_vllm_wheel_release ${COMMON_WORKDIR}/vllm/benchmarks /benchmarks COPY --from=build_vllm_wheel_release ${COMMON_WORKDIR}/vllm/tests /tests COPY --from=build_vllm_wheel_release ${COMMON_WORKDIR}/vllm/examples /examples +COPY --from=build_vllm_wheel_release ${COMMON_WORKDIR}/vllm/tools/install_torchcodec_rocm.sh /tools/install_torchcodec_rocm.sh COPY --from=build_vllm_wheel_release ${COMMON_WORKDIR}/vllm/docker/Dockerfile.rocm /docker/ COPY --from=build_vllm_wheel_release ${COMMON_WORKDIR}/vllm/.buildkite /.buildkite COPY --from=build_vllm_wheel_release ${COMMON_WORKDIR}/vllm/pyproject.toml /pyproject.toml diff --git a/requirements/rocm.txt b/requirements/rocm.txt index 4ca70738303..9dbf4bc96a3 100644 --- a/requirements/rocm.txt +++ b/requirements/rocm.txt @@ -11,7 +11,7 @@ numba == 0.65.0 # Required for N-gram speculative decoding datasets peft pytest-asyncio -tensorizer==2.10.1 +tensorizer==2.12.1 packaging>=24.2 setuptools>=77.0.3,<80.0.0 setuptools-scm>=8 diff --git a/requirements/test/cuda.in b/requirements/test/cuda.in index 344a58ec1bb..749b12d0460 100644 --- a/requirements/test/cuda.in +++ b/requirements/test/cuda.in @@ -1,6 +1,6 @@ # testing pytest -tensorizer==2.10.1 +tensorizer==2.12.1 pytest-forked pytest-asyncio pytest-rerunfailures diff --git a/requirements/test/cuda.txt b/requirements/test/cuda.txt index 7d847d10577..b36934d12d2 100644 --- a/requirements/test/cuda.txt +++ b/requirements/test/cuda.txt @@ -965,7 +965,7 @@ tenacity==9.1.2 # gpt-oss # lm-eval # plotly -tensorizer==2.10.1 +tensorizer==2.12.1 # via -r requirements/test/cuda.in termcolor==3.1.0 # via gpt-oss diff --git a/requirements/test/nightly-torch.txt b/requirements/test/nightly-torch.txt index 89fd4ea9b43..e9be984b360 100644 --- a/requirements/test/nightly-torch.txt +++ b/requirements/test/nightly-torch.txt @@ -1,6 +1,6 @@ # testing pytest -tensorizer==2.10.1 +tensorizer==2.12.1 pytest-forked pytest-asyncio pytest-rerunfailures diff --git a/requirements/test/rocm.in b/requirements/test/rocm.in index 0a615831774..38a37b837ed 100644 --- a/requirements/test/rocm.in +++ b/requirements/test/rocm.in @@ -2,7 +2,7 @@ # testing pytest -tensorizer==2.10.1 +tensorizer==2.12.1 pytest-forked pytest-asyncio pytest-rerunfailures diff --git a/requirements/test/rocm.txt b/requirements/test/rocm.txt index e0232d8b6d3..2ab2d979435 100644 --- a/requirements/test/rocm.txt +++ b/requirements/test/rocm.txt @@ -1186,7 +1186,7 @@ tenacity==9.1.4 # via # gpt-oss # lm-eval -tensorizer==2.10.1 +tensorizer==2.12.1 # via # -c requirements/rocm.txt # -r requirements/test/rocm.in