gnu: Add rocfft.

* gnu/packages/rocm-libs.scm (rocfft): New variable.

Signed-off-by: Ludovic Courtès <ludo@gnu.org>

1 files changed, 25 insertions, 0 deletions

diff --git a/gnu/packages/rocm-libs.scm b/gnu/packages/rocm-libs.scm
index 0d63415e0a3..a0f679a0d31 100644
--- a/gnu/packages/rocm-libs.scm
+++ b/gnu/packages/rocm-libs.scm
@@ -871,3 +871,28 @@ operations on this type.")
 machine learning primitives.  This package contains the HIP version based on
 ROCm.")
     (license license:expat)))
+
+(define-public rocfft
+  (package
+    (name "rocfft")
+    (version %rocm-version)
+    (source (rocm-library-source "rocfft"))
+    (build-system cmake-build-system)
+    (arguments
+     (list
+      #:tests? #f ; requires GPU
+      #:build-type "Release"
+      #:configure-flags
+      #~(list "-DCMAKE_CXX_COMPILER=hipcc"
+              "-DCMAKE_C_COMPILER=hipcc"
+              #$(string-append "-DAMDGPU_TARGETS="
+                               (current-amd-gpu-targets-string))
+              "-DSQLITE_USE_SYSTEM_PACKAGE=ON")))
+    (inputs (list rocm-hip-runtime sqlite-next))        ;needs sqlite >= 3.50.2
+    (native-inputs (list python rocm-cmake rocm-toolchain))
+    (properties `((amd-gpu-targets . ,%default-amd-gpu-targets)))
+    (home-page %rocm-libraries-url)
+    (synopsis "FFT library for the HIP programming language")
+    (description "This package contains a HIP library for computing fast
+Fourier transforms.")
+    (license license:expat)))