Skip to content

GitLab

"...models/git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "07fb8ba7fad7b5b458ff862919825df4e6f60b52"
Unverified Commit 718f25ae authored by Keyang Ru's avatar Keyang Ru Committed by GitHub
Browse files

Explicitly export CMAKE_BUILD_PARALLEL_LEVEL (#10193)

parent 948b01a0
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
sgl-kernel/build.sh
View file @ 718f25ae
...@@ -15,7 +15,6 @@ echo "ARCH: $ARCH" ...@@ -15,7 +15,6 @@ echo "ARCH: $ARCH"
if [ ${ARCH} = "aarch64" ]; then if [ ${ARCH} = "aarch64" ]; then
LIBCUDA_ARCH="sbsa" LIBCUDA_ARCH="sbsa"
BUILDER_NAME="pytorch/manylinuxaarch64-builder" BUILDER_NAME="pytorch/manylinuxaarch64-builder"
CMAKE_BUILD_PARALLEL_LEVEL=16
else else
LIBCUDA_ARCH=${ARCH} LIBCUDA_ARCH=${ARCH}
BUILDER_NAME="pytorch/manylinux2_28-builder" BUILDER_NAME="pytorch/manylinux2_28-builder"
...@@ -40,6 +39,7 @@ docker run --rm \ ...@@ -40,6 +39,7 @@ docker run --rm \
export CMAKE_VERSION_MAJOR=3.31 export CMAKE_VERSION_MAJOR=3.31
export CMAKE_VERSION_MINOR=1 export CMAKE_VERSION_MINOR=1
# Setting these flags to reduce OOM chance only on ARM # Setting these flags to reduce OOM chance only on ARM
export CMAKE_BUILD_PARALLEL_LEVEL=$(( $(nproc)/3 < 48 ? $(nproc)/3 : 48 ))
if [ \"${ARCH}\" = \"aarch64\" ]; then if [ \"${ARCH}\" = \"aarch64\" ]; then
export CUDA_NVCC_FLAGS=\"-Xcudafe --threads=2\" export CUDA_NVCC_FLAGS=\"-Xcudafe --threads=2\"
export MAKEFLAGS='-j2' export MAKEFLAGS='-j2'
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment