24 lines
623 B
YAML
24 lines
623 B
YAML
group: Hardware - AMD Build
|
|
steps:
|
|
- label: "AMD: :docker: build image"
|
|
key: image-build-amd
|
|
depends_on: []
|
|
device: amd_cpu
|
|
no_plugin: true
|
|
soft_fail: true
|
|
commands:
|
|
- >
|
|
docker build
|
|
--build-arg max_jobs=16
|
|
--build-arg REMOTE_VLLM=1
|
|
--build-arg ARG_PYTORCH_ROCM_ARCH='gfx90a;gfx942;gfx950'
|
|
--build-arg VLLM_BRANCH=$BUILDKITE_COMMIT
|
|
--tag "rocm/vllm-ci:${BUILDKITE_COMMIT}"
|
|
-f docker/Dockerfile.rocm
|
|
--target test
|
|
--no-cache
|
|
--progress plain .
|
|
- docker push "rocm/vllm-ci:${BUILDKITE_COMMIT}"
|
|
env:
|
|
DOCKER_BUILDKIT: "1"
|