Skip to content

Commit

Permalink
Fix CPU image build workflow
Browse files Browse the repository at this point in the history
  • Loading branch information
sd109 committed Feb 14, 2025
1 parent f40288e commit 6d18693
Showing 1 changed file with 71 additions and 0 deletions.
71 changes: 71 additions & 0 deletions .github/workflows/build-push-vllm-cpu.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,71 @@
name: Publish vLLM CPU images

on:
# NOTE(sd109): Since this is checking out an external
# it's probably safer to leave this as workflow dispatch
# only so that we can manually build images from specific
# refs rather than automatically pulling in the latest
# content from the remote repo.
workflow_dispatch:
inputs:
vllm_ref:
type: string
description: The vLLM GitHub ref (tag, branch or commit) to build.
required: true

jobs:
build_push_x86_image:
name: Build and push image
runs-on: ubuntu-latest
permissions:
contents: read
id-token: write # needed for signing the images with GitHub OIDC Token
packages: write # required for pushing container images
security-events: write # required for pushing SARIF files
steps:
- name: Check out the vLLM repository
uses: actions/checkout@v4
with:
repository: vllm-project/vllm
ref: ${{ inputs.vllm_ref }}

- name: Login to GitHub Container Registry
uses: docker/login-action@v3
with:
registry: ghcr.io
username: ${{ github.actor }}
password: ${{ secrets.GITHUB_TOKEN }}

- name: Build and push image
run: |
IMAGE=ghcr.io/stackhpc/vllm-cpu:${{ inputs.vllm_ref }}
docker build -f Dockerfile.cpu -t $IMAGE --shm-size=4g .
docker push $IMAGE
build_push_arm64_image:
name: Build and push image
runs-on: ubuntu-24.04-arm
permissions:
contents: read
id-token: write # needed for signing the images with GitHub OIDC Token
packages: write # required for pushing container images
security-events: write # required for pushing SARIF files
steps:
- name: Check out the vLLM repository
uses: actions/checkout@v4
with:
repository: vllm-project/vllm
ref: ${{ inputs.vllm_ref }}

- name: Login to GitHub Container Registry
uses: docker/login-action@v3
with:
registry: ghcr.io
username: ${{ github.actor }}
password: ${{ secrets.GITHUB_TOKEN }}

- name: Build and push image
run: |
IMAGE=ghcr.io/stackhpc/vllm-cpu:${{ inputs.vllm_ref }}-arm64
docker build -f Dockerfile.arm -t $IMAGE --shm-size=4g .
docker push $IMAGE

0 comments on commit 6d18693

Please sign in to comment.