1
+ account_id : &ACCOUNT_ID <set-$ACCOUNT_ID-in-environment>
2
+ prod_account_id : &PROD_ACCOUNT_ID 763104351884
3
+ region : ®ION <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
0 commit comments