diff --git a/.github/PULL_REQUEST_TEMPLATE.md b/.github/PULL_REQUEST_TEMPLATE.md index f16eca83ad8d..4aa3b8a8e886 100644 --- a/.github/PULL_REQUEST_TEMPLATE.md +++ b/.github/PULL_REQUEST_TEMPLATE.md @@ -1,11 +1,70 @@ -## PR Checklist +## Purpose -- [] 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). +## Test Plan + +## Test Result + +______________________________________________________________________ + +
+Toggle if you are merging into master Branch + +By default, docker image builds and tests are disabled. Two ways to run builds and tests: + +1. Using dlc_developer_config.toml +1. Using this PR description (currently only supported for PyTorch, TensorFlow, vllm, and base images) + +
+How to use the helper utility for updating dlc_developer_config.toml + +Assuming your remote is called `origin` (you can find out more with `git remote -v`)... + +- Run default builds and tests for a particular buildspec - also commits and pushes changes to remote; Example: + +`python src/prepare_dlc_dev_environment.py -b -cp origin` + +- Enable specific tests for a buildspec or set of buildspecs - also commits and pushes changes to remote; Example: + +`python src/prepare_dlc_dev_environment.py -b -t sanity_tests -cp origin` -## Description +- Restore TOML file when ready to merge - +`python src/prepare_dlc_dev_environment.py -rcp origin` -## Changes Made +**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:** + +- [ ] `sagemaker_remote_tests = true` +- [ ] `sagemaker_efa_tests = true` +- [ ] `sagemaker_rc_tests = true` +- [ ] `sagemaker_local_tests = true` + +
+ +
+How to use PR description +Use the code block below to uncomment commands and run the PR CodeBuild jobs. There are two commands available: + +- `# /buildspec ` + - e.g.: `# /buildspec pytorch/training/buildspec.yml` + - If this line is commented out, dlc_developer_config.toml will be used. +- `# /tests ` + - e.g.: `# /tests sanity security ec2` + - 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`. + +
+ +``` +/buildspec vllm/buildspec.yml +# /tests +``` + +
+ +
+Toggle if you are merging into main Branch + +## PR Checklist + +- [] 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). -## Testing Done +
diff --git a/README.md b/README.md index 82e3488470cf..07718dfe6915 100644 --- a/README.md +++ b/README.md @@ -4,7 +4,7 @@

One stop shop for running AI/ML on AWS

-

AWS Doc · +

Docs · Available Images · Tutorials

______________________________________________________________________ @@ -13,9 +13,9 @@ ______________________________________________________________________ ### 🚀 Release Highlights -- **[2025/12/19]** Released v0.13.0 [vLLM DLCs](https://gallery.ecr.aws/deep-learning-containers/vllm) - - EC2/EKS/ECS: `public.ecr.aws/deep-learning-containers/vllm:0.13-gpu-py312-cu130-ubuntu22.04-ec2` - - SageMaker: `public.ecr.aws/deep-learning-containers/vllm:0.13.0-gpu-py312` +- **[2026/02/06]** Released v0.15.1 [vLLM DLCs](https://gallery.ecr.aws/deep-learning-containers/vllm) + - EC2/EKS/ECS: `public.ecr.aws/deep-learning-containers/vllm:0.15-gpu-py312-ec2` + - SageMaker: `public.ecr.aws/deep-learning-containers/vllm:0.15-gpu-py312` - **[2025/11/17]** Released first [SGLang DLCs](https://gallery.ecr.aws/deep-learning-containers/sglang) - SageMaker: `public.ecr.aws/deep-learning-containers/sglang:0.5.5-gpu-py312` diff --git a/docker/vllm/Dockerfile b/docker/vllm/Dockerfile index ccae436861d1..773e7fe8a436 100644 --- a/docker/vllm/Dockerfile +++ b/docker/vllm/Dockerfile @@ -8,7 +8,7 @@ FROM $BASE_IMAGE AS base # ====================== common ========================================= ARG PYTHON="python3" LABEL maintainer="Amazon AI" -ARG EFA_VERSION="1.46.0" +ARG EFA_VERSION="1.47.0" LABEL dlc_major_version="1" ENV DEBIAN_FRONTEND=noninteractive \ LANG=C.UTF-8 \