fix arm sgl-kernel link issue (#7066)
This commit is contained in:
@@ -15,6 +15,7 @@ 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 +41,7 @@ docker run --rm \
|
|||||||
tar -xzf cmake-\${CMAKE_VERSION_MAJOR}.\${CMAKE_VERSION_MINOR}-linux-${ARCH}.tar.gz
|
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
|
mv cmake-\${CMAKE_VERSION_MAJOR}.\${CMAKE_VERSION_MINOR}-linux-${ARCH} /opt/cmake
|
||||||
export PATH=/opt/cmake/bin:\$PATH
|
export PATH=/opt/cmake/bin:\$PATH
|
||||||
|
export LD_LIBRARY_PATH=/lib64:\$LD_LIBRARY_PATH
|
||||||
|
|
||||||
# Debugging CMake
|
# Debugging CMake
|
||||||
echo \"PATH: \$PATH\"
|
echo \"PATH: \$PATH\"
|
||||||
|
|||||||
Reference in New Issue
Block a user