94 lines
3.1 KiB
YAML
94 lines
3.1 KiB
YAML
name: Release Docker Images (NPU)
|
|
on:
|
|
push:
|
|
tags:
|
|
- 'v[0-9]+.*'
|
|
workflow_dispatch:
|
|
inputs:
|
|
version:
|
|
description: 'Version to build (without v prefix, e.g., 0.5.7)'
|
|
required: true
|
|
|
|
jobs:
|
|
build:
|
|
runs-on: ubuntu-22.04-arm
|
|
strategy:
|
|
matrix:
|
|
cann_version: ["8.5.0"]
|
|
device_type: ["910b", "a3"]
|
|
steps:
|
|
- name: Checkout repository
|
|
uses: actions/checkout@v4
|
|
|
|
- name: Free up disk space
|
|
uses: jlumbroso/free-disk-space@54081f138730dfa15788a46383842cd2f914a1be # v1.3.1
|
|
with:
|
|
tool-cache: true
|
|
docker-images: false
|
|
|
|
# push with tag
|
|
- name: Docker meta
|
|
id: meta
|
|
uses: docker/metadata-action@v5
|
|
with:
|
|
images: |
|
|
lmsysorg/sglang
|
|
tags: |
|
|
type=ref,event=pr
|
|
flavor: |
|
|
latest=false
|
|
|
|
# Login against a Docker registry except on PR
|
|
# https://github.com/docker/login-action
|
|
- name: Login to Docker Hub
|
|
uses: docker/login-action@v2
|
|
if: ${{ github.repository == 'sgl-project/sglang' && github.event_name != 'pull_request' }}
|
|
with:
|
|
username: ${{ secrets.DOCKERHUB_USERNAME }}
|
|
password: ${{ secrets.DOCKERHUB_TOKEN }}
|
|
|
|
- name: Get version from tag
|
|
id: version
|
|
run: |
|
|
if [ "${{ github.event_name }}" = "workflow_dispatch" ]; then
|
|
VERSION="${{ github.event.inputs.version }}"
|
|
else
|
|
# Extract version from tag (e.g., v0.5.7 -> 0.5.7)
|
|
VERSION="${GITHUB_REF_NAME#v}"
|
|
fi
|
|
# Validate version format
|
|
if [ -z "$VERSION" ]; then
|
|
echo "::error::Version is empty"
|
|
exit 1
|
|
fi
|
|
if ! echo "$VERSION" | grep -qE '^[0-9]+\.[0-9]+\.[0-9]+'; then
|
|
echo "::error::Invalid version format: $VERSION (expected: X.Y.Z)"
|
|
exit 1
|
|
fi
|
|
echo "version=v${VERSION}" >> $GITHUB_OUTPUT
|
|
echo "TAG=lmsysorg/sglang:v${VERSION}-cann${{ matrix.cann_version }}-${{ matrix.device_type }}" >> $GITHUB_OUTPUT
|
|
# Enable Docker multi-architecture build environment
|
|
# Emulate non-native architectures
|
|
- name: Set up QEMU
|
|
uses: docker/setup-qemu-action@v3
|
|
# Required for building and pushing multi-arch Docker images
|
|
- name: Set up Docker Buildx
|
|
uses: docker/setup-buildx-action@v3
|
|
|
|
- name: Build and push Docker image
|
|
id: build-and-push
|
|
uses: docker/build-push-action@v6
|
|
with:
|
|
context: docker
|
|
file: docker/npu.Dockerfile
|
|
platforms: linux/arm64,linux/amd64
|
|
labels: ${{ steps.meta.outputs.labels }}
|
|
tags: ${{ steps.meta.outputs.tags || steps.version.outputs.TAG }}
|
|
push: ${{ github.repository == 'sgl-project/sglang' && github.event_name != 'pull_request' }}
|
|
provenance: false
|
|
build-args: |
|
|
SGLANG_KERNEL_NPU_TAG=2026.03.10.rc1
|
|
CANN_VERSION=${{ matrix.cann_version }}
|
|
DEVICE_TYPE=${{ matrix.device_type }}
|
|
SGLANG_TAG=${{ steps.version.outputs.version }}
|