diff options
author | a.kudelin | 2020-01-08 13:04:33 +0300 |
---|---|---|
committer | a.kudelin | 2020-01-08 13:04:33 +0300 |
commit | 94eea3482d7bf1ea509b58b66e58fb04a13f0e34 (patch) | |
tree | f1e1d724c7b0faa8f3d180aca9bf003d49fd1f49 /PKGBUILD | |
download | aur-94eea3482d7bf1ea509b58b66e58fb04a13f0e34.tar.gz |
Initial commit
Diffstat (limited to 'PKGBUILD')
-rw-r--r-- | PKGBUILD | 50 |
1 files changed, 50 insertions, 0 deletions
diff --git a/PKGBUILD b/PKGBUILD new file mode 100644 index 000000000000..5cec3245ce64 --- /dev/null +++ b/PKGBUILD @@ -0,0 +1,50 @@ +# Maintainer: Anton Kudelin <kudelin at protonmail dot com> + +pkgname=spfft +_pkgname=SpFFT +pkgver=0.9.10 +pkgrel=1 +pkgdesc="Sparse 3D FFT library with MPI, OpenMP, CUDA and ROCm support" +arch=("x86_64") +url="https://github.com/eth-cscs/SpFFT" +license=('BSD') +depends=('fftw-mpi') +makedepends=('cmake') +optdepends=('cuda: GPU support') +provides=('spfft') +conflicts=('spfft-cuda-git') +source=("$url/archive/v$pkgver.tar.gz") +sha256sums=('b4a5cbdf69b5dd97202f269f24e2337b333a45d6ff001aabfca64c7ec412c15d') + +prepare() { + mkdir $srcdir/build + + # Checking if nvcc is in PATH + if [ $( echo -n $( which nvcc) | tail -c 4 ) == nvcc ] + then + export _ACC=CUDA + export LDFLAGS="$LDFLAGS -L/opt/cuda/lib64" + echo "GPU is enabled" + else + export _ACC=OFF + echo "GPU is disabled" + fi +} + +build() { + cd $srcdir/build + cmake ../$_pkgname-$pkgver \ + -DCMAKE_INSTALL_PREFIX=/usr \ + -DSPFFT_FORTRAN=ON \ + -DSPFFT_MPI=ON \ + -DSPFFT_OMP=ON \ + -DSPFFT_GPU_BACKEND=$_ACC + make CUDA_FLAGS="-O3 -Xcompiler=-fPIC" +} + +package() { + cd $srcdir/build + make DESTDIR=$pkgdir install + install -dm755 $pkgdir/usr/share/licenses/$pkgname + install ../$_pkgname-$pkgver/LICENSE $pkgdir/usr/share/licenses/$pkgname +} |