Skip to content

GitLab

"official/vision/examples/starter/example_model.py" did not exist on "c32ce7cfaac3a69b9944374aac41ea9135b1d1cc"
Unverified Commit 344adb00 authored by Yineng Zhang's avatar Yineng Zhang Committed by GitHub
Browse files

fix arm sgl-kernel link issue (#7066)

parent b56de8f9
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
sgl-kernel/build.sh
View file @ 344adb00
......@@ -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\"
......
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