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:
@@ -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"
|
||||
|
||||
Reference in New Issue
Block a user