diff options
Diffstat (limited to 'libraries/pytorch/pytorch.SlackBuild')
-rw-r--r-- | libraries/pytorch/pytorch.SlackBuild | 135 |
1 files changed, 135 insertions, 0 deletions
diff --git a/libraries/pytorch/pytorch.SlackBuild b/libraries/pytorch/pytorch.SlackBuild new file mode 100644 index 0000000000..78f4a0a68f --- /dev/null +++ b/libraries/pytorch/pytorch.SlackBuild @@ -0,0 +1,135 @@ +#!/bin/bash + +# Slackware build script for pytorch + +# Copyright 2024 Johannes Schoepfer, Germany +# All rights reserved. +# +# Redistribution and use of this script, with or without modification, is +# permitted provided that the following conditions are met: +# +# 1. Redistributions of this script must retain the above copyright +# notice, this list of conditions and the following disclaimer. +# +# THIS SOFTWARE IS PROVIDED BY THE AUTHOR "AS IS" AND ANY EXPRESS OR IMPLIED +# WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF +# MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO +# EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, +# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, +# PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; +# OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, +# WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR +# OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF +# ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + +cd $(dirname $0) ; CWD=$(pwd) + +PRGNAM=pytorch +VERSION=${VERSION:-2.4.0} +COMMIT_QNNPACK=7d2a4e9931a82adc3814275b6219a03e24e36b4c +BUILD=${BUILD:-1} +TAG=${TAG:-_SBo} +PKGTYPE=${PKGTYPE:-tgz} + +if [ -z "$ARCH" ]; then + case "$( uname -m )" in + i?86) ARCH=i586 ;; + arm*) ARCH=arm ;; + *) ARCH=$( uname -m ) ;; + esac +fi + +if [ ! -z "${PRINT_PACKAGE_NAME}" ]; then + echo "$PRGNAM-$VERSION-$ARCH-$BUILD$TAG.$PKGTYPE" + exit 0 +fi + +TMP=${TMP:-/tmp/SBo} +PKG=$TMP/package-$PRGNAM +OUTPUT=${OUTPUT:-/tmp} + +if [ "$ARCH" = "i586" ]; then + SLKCFLAGS="-O2 -march=i586 -mtune=i686" +elif [ "$ARCH" = "i686" ]; then + SLKCFLAGS="-O2 -march=i686 -mtune=i686" +elif [ "$ARCH" = "aarch64" ]; then + SLKCFLAGS="-O2 -fPIC" + LIBDIRSUFFIX="64" +elif [ "$ARCH" = "x86_64" ]; then + SLKCFLAGS="-O2 -fPIC" + LIBDIRSUFFIX="64" +elif [ "$ARCH" = "arm" ]; then + SLKCFLAGS="-O2 -march=armv7-a -mfpu=vfpv3-d16 -mfloat-abi=hard" +else + SLKCFLAGS="-O2" +fi + +set -e + +rm -rf $PKG +mkdir -p $TMP $PKG $OUTPUT +cd $TMP +rm -rf $PRGNAM-v$VERSION +tar xvf $CWD/$PRGNAM-v$VERSION.tar.gz +cd $PRGNAM-v$VERSION +chown -R root:root . +chmod -R u+w,go-w,a+rX-st $PKG + +# QNNPACK not present in tar ball since version 2.4.0 +( +cd third_party +tar vxf $CWD/QNNPACK-$COMMIT_QNNPACK.tar.gz +ln -s QNNPACK-$COMMIT_QNNPACK QNNPACK +) + +# Fix cmake libdir location +sed -i c10/CMakeLists.txt \ + -e "s;DESTINATION lib;DESTINATION lib$LIBDIRSUFFIX;g" + +TMP_DIR=tmpxxx +mkdir build +cd build + unshare -n cmake \ + -G Ninja \ + -DCMAKE_C_FLAGS:STRING="$SLKCFLAGS" \ + -DCMAKE_CXX_FLAGS:STRING="$SLKCFLAGS" \ + -DCMAKE_CXX_STANDARD=17 \ + -DCMAKE_INSTALL_PREFIX="/usr" \ + -DLIBSHM_INSTALL_LIB_SUBDIR="lib$LIBDIRSUFFIX" \ + -DTORCH_INSTALL_LIB_DIR="lib$LIBDIRSUFFIX" \ + -DPYTHON_EXECUTABLE=$(which python3) \ + -DBUILD_CUSTOM_PROTOBUF=OFF \ + -DBUILD_TEST=OFF \ + -DUSE_GOLD_LINKER=ON \ + -DUSE_OPENCL=ON \ + -DUSE_VULKAN=ON \ + -DCMAKE_BUILD_TYPE=Release .. + "${NINJA:=ninja}" + DESTDIR=$TMP_DIR $NINJA install/strip + + mkdir -p $PKG/usr/{share,lib$LIBDIRSUFFIX} + find $TMP_DIR -empty -type d -delete + mv $TMP_DIR/usr/bin $PKG/usr + mv $TMP_DIR/usr/include $PKG/usr + mv $TMP_DIR/usr/share/cmake $PKG/usr/share + mv $TMP_DIR/usr/lib$LIBDIRSUFFIX/*.so $PKG/usr/lib$LIBDIRSUFFIX +cd .. +python3 setup.py install --root=$PKG + +SITE="$(python3 -c 'import sysconfig; print(sysconfig.get_paths()["purelib"])')" +mkdir -p $PKG/$SITE/torch/{bin,lib} +BIN=torch_shm_manager; ln -s /usr/bin/$BIN $PKG/$SITE/torch/bin/$BIN +LIB=libtorch_global_deps.so; ln -s /usr/lib$LIBDIRSUFFIX/$LIB $PKG/$SITE/torch/lib/$LIB + +find $PKG -print0 | xargs -0 file | grep -e "executable" -e "shared object" | grep ELF \ + | cut -f 1 -d : | xargs strip --strip-unneeded 2> /dev/null || true + +mkdir -p $PKG/usr/doc/$PRGNAM-$VERSION +cp -a LICENSE NOTICE README.md RELEASE.md $PKG/usr/doc/$PRGNAM-$VERSION +cat $CWD/$PRGNAM.SlackBuild > $PKG/usr/doc/$PRGNAM-$VERSION/$PRGNAM.SlackBuild + +mkdir -p $PKG/install +cat $CWD/slack-desc > $PKG/install/slack-desc + +cd $PKG +/sbin/makepkg -l y -c n $OUTPUT/$PRGNAM-$VERSION-$ARCH-$BUILD$TAG.$PKGTYPE |