Skip to content

Commit eb25cb6

Browse files
committed
[vllm] [cpu] [sagemaker] Fix image size baseline and add ulimit
- Set image_size_baseline to 5000 (actual image ~3.5GB) - Add ulimit -c 0 to disable core dumps (matches upstream)
1 parent f921a5e commit eb25cb6

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

vllm/buildspec-cpu-sm.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ images:
3030
<<: *BUILD_REPOSITORY
3131
context:
3232
<<: *BUILD_CONTEXT
33-
image_size_baseline: 15000
33+
image_size_baseline: 5000
3434
device_type: &DEVICE_TYPE cpu
3535
python_version: &DOCKER_PYTHON_VERSION py3
3636
tag_python_version: &TAG_PYTHON_VERSION py312

vllm/x86_64/cpu/Dockerfile.cpu

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,8 @@ ENV UV_HTTP_TIMEOUT=500
6363
# Memory allocator + Intel OpenMP for x86_64 performance
6464
ENV LD_PRELOAD="/usr/lib/x86_64-linux-gnu/libtcmalloc_minimal.so.4:/opt/venv/lib/libiomp5.so"
6565

66+
RUN echo 'ulimit -c 0' >> ~/.bashrc
67+
6668
# ====================== vllm-build =========================================
6769
FROM base AS vllm-build
6870

0 commit comments

Comments
 (0)