Skip to content

Commit 1e8ba12

Browse files
add buildspec for vllm arm64 (#5141)
1 parent c2eee62 commit 1e8ba12

File tree

3 files changed

+53
-1
lines changed

3 files changed

+53
-1
lines changed

vllm/buildspec-arm64.yml

Lines changed: 52 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,52 @@
1+
account_id: &ACCOUNT_ID <set-$ACCOUNT_ID-in-environment>
2+
prod_account_id: &PROD_ACCOUNT_ID 763104351884
3+
region: &REGION <set-$REGION-in-environment>
4+
framework: &FRAMEWORK vllm
5+
version: &VERSION "0.10.0"
6+
short_version: &SHORT_VERSION "0.10"
7+
arch_type: &ARCH_TYPE arm64
8+
autopatch_build: "False"
9+
10+
repository_info:
11+
build_repository: &BUILD_REPOSITORY
12+
image_type: &IMAGE_TYPE gpu
13+
root: .
14+
repository_name: &REPOSITORY_NAME !join [ pr, "-", *FRAMEWORK ]
15+
repository: &REPOSITORY !join [ *ACCOUNT_ID, .dkr.ecr., *REGION, .amazonaws.com/, *REPOSITORY_NAME ]
16+
release_repository_name: &RELEASE_REPOSITORY_NAME !join [ *FRAMEWORK ]
17+
release_repository: &RELEASE_REPOSITORY !join [ *PROD_ACCOUNT_ID, .dkr.ecr., *REGION, .amazonaws.com/, *RELEASE_REPOSITORY_NAME ]
18+
19+
context:
20+
build_context: &BUILD_CONTEXT
21+
deep_learning_container:
22+
source: src/deep_learning_container.py
23+
target: deep_learning_container.py
24+
install_efa:
25+
source: scripts/install_efa.sh
26+
target: install_efa.sh
27+
dockerd_entrypoint:
28+
source: vllm/build_artifacts/dockerd_entrypoint.sh
29+
target: dockerd_entrypoint.sh
30+
31+
images:
32+
vllm_arm64:
33+
<<: *BUILD_REPOSITORY
34+
context:
35+
<<: *BUILD_CONTEXT
36+
image_size_baseline: 20000
37+
device_type: &DEVICE_TYPE gpu
38+
cuda_version: &CUDA_VERSION cu128
39+
python_version: &DOCKER_PYTHON_VERSION py3
40+
tag_python_version: &TAG_PYTHON_VERSION py312
41+
os_version: &OS_VERSION ubuntu22.04
42+
tag: !join [ *VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *CUDA_VERSION, "-", *OS_VERSION, "-ec2" ]
43+
latest_release_tag: !join [ *VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *CUDA_VERSION, "-", *OS_VERSION, "-ec2" ]
44+
docker_file: !join [ *FRAMEWORK, /, *ARCH_TYPE, /, *DEVICE_TYPE, /Dockerfile ]
45+
target: final
46+
build: true
47+
enable_common_stage_build: false
48+
test_configs:
49+
test_platforms:
50+
- sanity
51+
- security
52+
- eks

vllm/buildspec.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ context:
2525
source: scripts/install_efa.sh
2626
target: install_efa.sh
2727
dockerd_entrypoint:
28-
source: vllm/x86_64/build_artifacts/dockerd_entrypoint.sh
28+
source: vllm/build_artifacts/dockerd_entrypoint.sh
2929
target: dockerd_entrypoint.sh
3030

3131
images:

0 commit comments

Comments
 (0)