ci: improve docker for cu13 builds (#18194)

This commit is contained in:
ishandhanani
2026-02-03 13:09:38 -06:00
committed by GitHub
parent 0db6fd4dbe
commit 0a6925639b
2 changed files with 7 additions and 3 deletions

View File

@@ -9,6 +9,10 @@ on:
version:
description: "Version to build (without v prefix, e.g., 0.5.8)"
required: true
flashinfer_version:
description: "FlashInfer version (default: 0.6.1)"
required: false
default: "0.6.1"
jobs:
publish-x86:
@@ -68,6 +72,7 @@ jobs:
--build-arg CUDA_VERSION=13.0.1 \
--build-arg BUILD_TYPE=all \
--build-arg INSTALL_FLASHINFER_JIT_CACHE=1 \
--build-arg FLASHINFER_VERSION=${{ github.event.inputs.flashinfer_version }} \
--build-arg GRACE_BLACKWELL=0 \
--build-arg SGL_VERSION=${version} \
-t lmsysorg/sglang:${tag} \
@@ -120,6 +125,7 @@ jobs:
--build-arg CUDA_VERSION=13.0.1 \
--build-arg BUILD_TYPE=all \
--build-arg INSTALL_FLASHINFER_JIT_CACHE=1 \
--build-arg FLASHINFER_VERSION=${{ github.event.inputs.flashinfer_version }} \
--build-arg GRACE_BLACKWELL=1 \
--build-arg SGL_VERSION=${version} \
-t lmsysorg/sglang:${tag} \