#!/bin/sh

# Slackware build script for Magma HIP

# Written by Condor

set -e

PRGNAM=magma
VERSION=2.9.0
BUILD=${BUILD:-1}
TAG=${TAG:-condor}
CWD=$(pwd)
TMP=${TMP:-/tmp/cnt}
PKG=$TMP/package-$PRGNAM
OUTPUT=${OUTPUT:-/tmp}
NUMJOBS=${NUMJOBS:-" -j$(expr $(nproc) + 1) "}

if [ -z "$ARCH" ]; then
  case "$( uname -m )" in
    i?86) ARCH=i486 ;;
    arm*) ARCH=arm ;;
       *) ARCH=$( uname -m ) ;;
  esac
fi

if [ "$ARCH" = "i486" ]; then
  SLKCFLAGS="-O2 -march=i486 -mtune=i686"
  LIBDIRSUFFIX=""
elif [ "$ARCH" = "i686" ]; then
  SLKCFLAGS="-O2 -march=i686 -mtune=i686"
  LIBDIRSUFFIX=""
elif [ "$ARCH" = "x86_64" ]; then
  SLKCFLAGS="-O2 -fPIC"
  LIBDIRSUFFIX="64"
else
  SLKCFLAGS="-O2"
  LIBDIRSUFFIX=""
fi

rm -rf $PKG
mkdir -p $TMP $PKG $OUTPUT
cd $TMP
rm -rf $PRGNAM-$VERSION
tar -xf $CWD/$PRGNAM-$VERSION.tar.gz
cd $PRGNAM-$VERSION
patch -Np1 -i "$CWD/add_cuda_archs.patch"
patch -Np1 -i "$CWD/hip_device.patch"
patch -Np1 -i "$CWD/min_max_macros.patch"

chown -R root:root .
chmod -R u+w,go+r-w,a-s .

# Fix cmake libdir location
sed -i CMakeLists.txt \
  -e "s;DESTINATION lib;DESTINATION lib$LIBDIRSUFFIX;g"
sed -i CMakeLists.txt \
  -e "s;DESTINATION lib/pkgconfig;DESTINATION lib$LIBDIRSUFFIX/pkgconfig;g"

_rocm_ver=$(./tools/get-rocm-version.sh)
CXXFLAGS+=" -fcf-protection=none"
CXXFLAGS+=" -DROCM_VERSION=$_rocm_ver"
CXXFLAGS+=" -isystem /opt/rocm/include/hipsparse"

_valid_gfx() {
  #List of GPU targets from rocBLAS
  echo "gfx900 gfx906:xnack- gfx908:xnack- gfx90a:xnack+ gfx90a:xnack- gfx940 gfx941 gfx942 gfx1010 gfx1012 gfx1030 gfx1100 gfx1101 gfx1102 gfx1115"
}


mkdir build
cd build

#CFLAGS="$SLKCFLAGS" \
#CXXFLAGS="$SLKCFLAGS" \
cmake \
  -D MAGMA_ENABLE_HIP=ON \
  -D CMAKE_CXX_COMPILER=/opt/rocm/bin/hipcc \
  -D CMAKE_BUILD_TYPE=Release \
  -D CMAKE_INSTALL_PREFIX=/opt/rocm \
  -D BLA_VENDOR=OpenBLAS \
  -D BUILD_SHARED_LIBS=ON \
  -D MAGMA_ENABLE_HIP=ON \
  -D GPU_TARGET="$(_valid_gfx)" \
  -D AMDGPU_TARGETS="$(_valid_gfx | tr ' ' ';')" \
  -D CMAKE_INSTALL_PREFIX=/usr \
  ..

make $NUMJOBS
make install/strip DESTDIR=$PKG

mkdir -p $PKG/install
cat $CWD/slack-desc > $PKG/install/slack-desc
cat $CWD/magma.pc > $PKG/usr/lib64/pkgconfig/magma.pc

cd $PKG
/sbin/makepkg -l y -c n $OUTPUT/$PRGNAM-$VERSION-$ARCH-$BUILD$TAG.txz