Initial commit

2 files changed, 67 insertions, 0 deletions

diff --git a/.SRCINFO b/.SRCINFO
new file mode 100644
index 000000000000..95c315845a50
--- /dev/null
+++ b/.SRCINFO
@@ -0,0 +1,17 @@
+pkgbase = spfft
+	pkgdesc = Sparse 3D FFT library with MPI, OpenMP, CUDA and ROCm support
+	pkgver = 0.9.10
+	pkgrel = 1
+	url = https://github.com/eth-cscs/SpFFT
+	arch = x86_64
+	license = BSD
+	makedepends = cmake
+	depends = fftw-mpi
+	optdepends = cuda: GPU support
+	provides = spfft
+	conflicts = spfft-cuda-git
+	source = https://github.com/eth-cscs/SpFFT/archive/v0.9.10.tar.gz
+	sha256sums = b4a5cbdf69b5dd97202f269f24e2337b333a45d6ff001aabfca64c7ec412c15d
+
+pkgname = spfft
+
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
+}