From 344adb00ecbaed3db367352429e2665064a2f04b Mon Sep 17 00:00:00 2001 From: Yineng Zhang Date: Tue, 10 Jun 2025 15:23:22 -0700 Subject: [PATCH] fix arm sgl-kernel link issue (#7066) --- sgl-kernel/build.sh | 2 ++ 1 file changed, 2 insertions(+) diff --git a/sgl-kernel/build.sh b/sgl-kernel/build.sh index d11027157..b2c8c4c5e 100755 --- a/sgl-kernel/build.sh +++ b/sgl-kernel/build.sh @@ -15,6 +15,7 @@ echo "ARCH: $ARCH" if [ ${ARCH} = "aarch64" ]; then LIBCUDA_ARCH="sbsa" BUILDER_NAME="pytorch/manylinuxaarch64-builder" + CMAKE_BUILD_PARALLEL_LEVEL=16 else LIBCUDA_ARCH=${ARCH} BUILDER_NAME="pytorch/manylinux2_28-builder" @@ -40,6 +41,7 @@ docker run --rm \ tar -xzf cmake-\${CMAKE_VERSION_MAJOR}.\${CMAKE_VERSION_MINOR}-linux-${ARCH}.tar.gz mv cmake-\${CMAKE_VERSION_MAJOR}.\${CMAKE_VERSION_MINOR}-linux-${ARCH} /opt/cmake export PATH=/opt/cmake/bin:\$PATH + export LD_LIBRARY_PATH=/lib64:\$LD_LIBRARY_PATH # Debugging CMake echo \"PATH: \$PATH\"