ci: improve docker for cu13 builds (#18194)
This commit is contained in:
@@ -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} \
|
||||
|
||||
Reference in New Issue
Block a user