Skip to content

Commit f5a9e22

Browse files
authored
[v2] update vllm EFA version (#5669)
1 parent 29d1eda commit f5a9e22

File tree

3 files changed

+70
-11
lines changed

3 files changed

+70
-11
lines changed

.github/PULL_REQUEST_TEMPLATE.md

Lines changed: 65 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,70 @@
1-
## PR Checklist
1+
## Purpose
22

3-
- [] I ran `pre-commit run --all-files` locally before creating this PR. (Read [DEVELOPMENT.md](https://github.com/aws/deep-learning-containers/blob/main/DEVELOPMENT.md) for details).
3+
## Test Plan
4+
5+
## Test Result
6+
7+
______________________________________________________________________
8+
9+
<details>
10+
<summary>Toggle if you are merging into master Branch</summary>
11+
12+
By default, docker image builds and tests are disabled. Two ways to run builds and tests:
13+
14+
1. Using dlc_developer_config.toml
15+
1. Using this PR description (currently only supported for PyTorch, TensorFlow, vllm, and base images)
16+
17+
<details>
18+
<summary>How to use the helper utility for updating dlc_developer_config.toml</summary>
19+
20+
Assuming your remote is called `origin` (you can find out more with `git remote -v`)...
21+
22+
- Run default builds and tests for a particular buildspec - also commits and pushes changes to remote; Example:
23+
24+
`python src/prepare_dlc_dev_environment.py -b </path/to/buildspec.yml> -cp origin`
25+
26+
- Enable specific tests for a buildspec or set of buildspecs - also commits and pushes changes to remote; Example:
27+
28+
`python src/prepare_dlc_dev_environment.py -b </path/to/buildspec.yml> -t sanity_tests -cp origin`
429

5-
## Description
30+
- Restore TOML file when ready to merge
631

7-
<!-- Briefly describe the changes in this PR -->
32+
`python src/prepare_dlc_dev_environment.py -rcp origin`
833

9-
## Changes Made
34+
**NOTE: If you are creating a PR for a new framework version, please ensure success of the local, standard, rc, and efa sagemaker tests by updating the dlc_developer_config.toml file:**
35+
36+
- [ ] `sagemaker_remote_tests = true`
37+
- [ ] `sagemaker_efa_tests = true`
38+
- [ ] `sagemaker_rc_tests = true`
39+
- [ ] `sagemaker_local_tests = true`
40+
41+
</details>
42+
43+
<details>
44+
<summary>How to use PR description</summary>
45+
Use the code block below to uncomment commands and run the PR CodeBuild jobs. There are two commands available:
46+
47+
- `# /buildspec <buildspec_path>`
48+
- e.g.: `# /buildspec pytorch/training/buildspec.yml`
49+
- If this line is commented out, dlc_developer_config.toml will be used.
50+
- `# /tests <test_list>`
51+
- e.g.: `# /tests sanity security ec2`
52+
- If this line is commented out, it will run the default set of tests (same as the defaults in dlc_developer_config.toml): `sanity, security, ec2, ecs, eks, sagemaker, sagemaker-local`.
53+
54+
</details>
55+
56+
```
57+
/buildspec vllm/buildspec.yml
58+
# /tests <test_list>
59+
```
60+
61+
</details>
62+
63+
<details>
64+
<summary>Toggle if you are merging into main Branch</summary>
65+
66+
## PR Checklist
67+
68+
- [] I ran `pre-commit run --all-files` locally before creating this PR. (Read [DEVELOPMENT.md](https://github.com/aws/deep-learning-containers/blob/main/DEVELOPMENT.md) for details).
1069

11-
## Testing Done
70+
</details>

README.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44

55
<p align="center"><strong>One stop shop for running AI/ML on AWS</strong></p>
66

7-
<p align="center"><a href="https://aws.amazon.com/ai/machine-learning/containers/"><strong>AWS Doc</strong></a> ·
7+
<p align="center"><a href="https://aws.github.io/deep-learning-containers/"><strong>Docs</strong></a> ·
88
<a href="https://aws.github.io/deep-learning-containers/reference/available_images/"><strong>Available Images</strong></a> · <a href="https://aws.github.io/deep-learning-containers/tutorials/"><strong>Tutorials</strong></a></p>
99

1010
______________________________________________________________________
@@ -13,9 +13,9 @@ ______________________________________________________________________
1313

1414
### 🚀 Release Highlights
1515

16-
- **[2025/12/19]** Released v0.13.0 [vLLM DLCs](https://gallery.ecr.aws/deep-learning-containers/vllm)
17-
- EC2/EKS/ECS: `public.ecr.aws/deep-learning-containers/vllm:0.13-gpu-py312-cu130-ubuntu22.04-ec2`
18-
- SageMaker: `public.ecr.aws/deep-learning-containers/vllm:0.13.0-gpu-py312`
16+
- **[2026/02/06]** Released v0.15.1 [vLLM DLCs](https://gallery.ecr.aws/deep-learning-containers/vllm)
17+
- EC2/EKS/ECS: `public.ecr.aws/deep-learning-containers/vllm:0.15-gpu-py312-ec2`
18+
- SageMaker: `public.ecr.aws/deep-learning-containers/vllm:0.15-gpu-py312`
1919
- **[2025/11/17]** Released first [SGLang DLCs](https://gallery.ecr.aws/deep-learning-containers/sglang)
2020
- SageMaker: `public.ecr.aws/deep-learning-containers/sglang:0.5.5-gpu-py312`
2121

docker/vllm/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ FROM $BASE_IMAGE AS base
88
# ====================== common =========================================
99
ARG PYTHON="python3"
1010
LABEL maintainer="Amazon AI"
11-
ARG EFA_VERSION="1.46.0"
11+
ARG EFA_VERSION="1.47.0"
1212
LABEL dlc_major_version="1"
1313
ENV DEBIAN_FRONTEND=noninteractive \
1414
LANG=C.UTF-8 \

0 commit comments

Comments
 (0)