feat: add release workflow for SGLang kernels on aarch64 (#6010)

Co-authored-by: Qiaolin-Yu <liin1211@outlook.com>
Co-authored-by: Yineng Zhang <me@zhyncs.com>
This commit is contained in:
Johnny
2025-05-07 04:42:07 +02:00
committed by GitHub
parent d25398cbc8
commit cb69194562
2 changed files with 102 additions and 1 deletions

View File

@@ -5,7 +5,12 @@ PYTHON_VERSION=$1
CUDA_VERSION=$2
PYTHON_ROOT_PATH=/opt/python/cp${PYTHON_VERSION//.}-cp${PYTHON_VERSION//.}
ARCH=$(uname -i)
if [ -z "$3" ]; then
ARCH=$(uname -i)
else
ARCH=$3
fi
echo "ARCH: $ARCH"
if [ ${ARCH} = "aarch64" ]; then
LIBCUDA_ARCH="sbsa"