Skip to content

Commit d7ad54c

Browse files
test sm
1 parent db80d06 commit d7ad54c

File tree

2 files changed

+2
-24
lines changed

2 files changed

+2
-24
lines changed

vllm/buildspec-sm.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ context:
2929
target: sagemaker_entrypoint.sh
3030

3131
images:
32-
vllm_x86_64:
32+
vllm_sm:
3333
<<: *BUILD_REPOSITORY
3434
context:
3535
<<: *BUILD_CONTEXT

vllm/buildspec.yml

Lines changed: 1 addition & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -53,26 +53,4 @@ images:
5353
- sanity
5454
- security
5555
- ec2
56-
- eks
57-
vllm_sm:
58-
<<: *BUILD_REPOSITORY
59-
context:
60-
<<: *BUILD_CONTEXT
61-
image_size_baseline: 20000
62-
device_type: &DEVICE_TYPE gpu
63-
cuda_version: &CUDA_VERSION cu129
64-
python_version: &DOCKER_PYTHON_VERSION py3
65-
tag_python_version: &TAG_PYTHON_VERSION py312
66-
os_version: &OS_VERSION ubuntu22.04
67-
tag: !join [ *VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *CUDA_VERSION, "-", *OS_VERSION, "-sm" ]
68-
latest_release_tag: !join [ *VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *CUDA_VERSION, "-", *OS_VERSION, "-sm" ]
69-
docker_file: !join [ *FRAMEWORK, /, *ARCH_TYPE, /, *DEVICE_TYPE, /Dockerfile ]
70-
target: vllm-sagemaker
71-
build: true
72-
enable_common_stage_build: false
73-
test_configs:
74-
test_platforms:
75-
- sanity
76-
- security
77-
- sagemaker
78-
- eks
56+
- eks

0 commit comments

Comments
 (0)