Skip to content

Commit edba96b

Browse files
authored
fix(docker): migrate ROCm Dockerfiles from setuptools-rust to maturin (sgl-project#19210)
Signed-off-by: Simo Lin <linsimo.mark@gmail.com>
1 parent ea7ef63 commit edba96b

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

docker/rocm.Dockerfile

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -214,10 +214,10 @@ RUN curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y \
214214
ENV CARGO_BUILD_JOBS=4
215215

216216
# Build and install sgl-model-gateway
217-
RUN python3 -m pip install --no-cache-dir setuptools-rust \
217+
RUN python3 -m pip install --no-cache-dir maturin \
218218
&& cd /sgl-workspace/sglang/sgl-model-gateway/bindings/python \
219-
&& /bin/bash -lc 'ulimit -n 8192 && cargo build --release' \
220-
&& python3 -m pip install --no-cache-dir . \
219+
&& ulimit -n 65536 && maturin build --release --features vendored-openssl --out dist \
220+
&& python3 -m pip install --force-reinstall dist/*.whl \
221221
&& rm -rf /root/.cache
222222

223223
# -----------------------

docker/rocm720.Dockerfile

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -243,10 +243,10 @@ RUN curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y \
243243
ENV CARGO_BUILD_JOBS=4
244244

245245
# Build and install sgl-model-gateway
246-
RUN python3 -m pip install --no-cache-dir setuptools-rust \
246+
RUN python3 -m pip install --no-cache-dir maturin \
247247
&& cd /sgl-workspace/sglang/sgl-model-gateway/bindings/python \
248-
&& /bin/bash -lc 'ulimit -n 8192 && cargo build --release' \
249-
&& python3 -m pip install --no-cache-dir . \
248+
&& ulimit -n 65536 && maturin build --release --features vendored-openssl --out dist \
249+
&& python3 -m pip install --force-reinstall dist/*.whl \
250250
&& rm -rf /root/.cache
251251

252252
# -----------------------

0 commit comments

Comments
 (0)