From b8eba1c677a7dbd77b82ef4d4fdb3f620f77ed90 Mon Sep 17 00:00:00 2001 From: Weiqun Zhang Date: Mon, 18 Nov 2024 11:35:00 -0800 Subject: amrex: add new variant fft for >= 24.11 (#47611) --- var/spack/repos/builtin/packages/amrex/package.py | 7 +++++++ 1 file changed, 7 insertions(+) (limited to 'var') diff --git a/var/spack/repos/builtin/packages/amrex/package.py b/var/spack/repos/builtin/packages/amrex/package.py index 07c4d2ae66..b524b6af81 100644 --- a/var/spack/repos/builtin/packages/amrex/package.py +++ b/var/spack/repos/builtin/packages/amrex/package.py @@ -135,6 +135,7 @@ class Amrex(CMakePackage, CudaPackage, ROCmPackage): ) variant("eb", default=True, description="Build Embedded Boundary classes", when="@24.10:") variant("eb", default=False, description="Build Embedded Boundary classes", when="@:24.09") + variant("fft", default=False, description="Build FFT support", when="@24.11:") variant("fortran", default=False, description="Build Fortran API") variant("linear_solvers", default=True, description="Build linear solvers") variant("amrdata", default=False, description="Build data services") @@ -150,6 +151,9 @@ class Amrex(CMakePackage, CudaPackage, ROCmPackage): # Build dependencies depends_on("mpi", when="+mpi") + with when("+fft"): + depends_on("rocfft", when="+rocm") + depends_on("fftw@3", when="~cuda ~rocm ~sycl") with when("+ascent"): depends_on("ascent") depends_on("ascent +cuda", when="+cuda") @@ -330,6 +334,9 @@ class Amrex(CMakePackage, CudaPackage, ROCmPackage): self.define_from_variant("AMReX_PIC", "pic"), ] + if self.spec.satisfies("+fft"): + args.append("-DAMReX_FFT=ON") + if self.spec.satisfies("%fj"): args.append("-DCMAKE_Fortran_MODDIR_FLAG=-M") -- cgit v1.2.3-70-g09d2