cp2k: enable linking with armpl-gcc for BLAS and FFT (#34981)
This commit is contained in:
parent
c0389ab3d0
commit
10d5f14c85
@ -116,6 +116,7 @@ class Cp2k(MakefilePackage, CudaPackage):
|
||||
depends_on("fftw+openmp", when="^fftw")
|
||||
depends_on("amdfftw+openmp", when="^amdfftw")
|
||||
depends_on("cray-fftw+openmp", when="^cray-fftw")
|
||||
depends_on("armpl-gcc threads=openmp", when="^armpl-gcc")
|
||||
depends_on("openblas threads=openmp", when="^openblas")
|
||||
# The Cray compiler wrappers will automatically add libsci_mp with
|
||||
# -fopenmp. Since CP2K unconditionally links blas/lapack/scalapack
|
||||
@ -276,6 +277,9 @@ def edit(self, spec, prefix):
|
||||
elif "^amdfftw" in spec:
|
||||
fftw = spec["amdfftw:openmp" if "+openmp" in spec else "amdfftw"]
|
||||
fftw_header_dir = fftw.headers.directories[0]
|
||||
elif "^armpl-gcc" in spec:
|
||||
fftw = spec["armpl-gcc:openmp" if "+openmp" in spec else "armpl-gcc"]
|
||||
fftw_header_dir = fftw.headers.directories[0]
|
||||
elif "^intel-mkl" in spec:
|
||||
fftw = spec["intel-mkl"]
|
||||
fftw_header_dir = fftw.headers.directories[0] + "/fftw"
|
||||
|
Loading…
Reference in New Issue
Block a user