treewide: make 'gpuBackend' use 'config.(cuda|rocm)Support'

This commit is contained in:
Madoura 2023-11-14 08:41:58 -06:00
parent 242544ca17
commit 9709e96e1e
No known key found for this signature in database
GPG Key ID: 1B9BB76A2B71922D
4 changed files with 33 additions and 5 deletions

View File

@ -28,8 +28,16 @@
, spfft
, enableElpa ? false
, elpa
, gpuBackend ? "none"
, cudaPackages
, rocmPackages
, config
, gpuBackend ? (
if config.cudaSupport
then "cuda"
else if config.rocmSupport
then "rocm"
else "none"
)
# gpuVersion needs to be set for both CUDA as well as ROCM hardware.
# gpuArch is only required for the ROCM stack.
# Change to a value suitable for your target GPU.
@ -37,7 +45,6 @@
# and for Nvidia see https://github.com/cp2k/cp2k/blob/master/INSTALL.md#2i-cuda-optional-improved-performance-on-gpu-systems
, gpuVersion ? "Mi100"
, gpuArch ? "gfx908"
, rocmPackages
}:
assert builtins.elem gpuBackend [ "none" "cuda" "rocm" ];

View File

@ -21,9 +21,16 @@
, eigen
, libvdwxc
, llvmPackages
, gpuBackend ? "none"
, cudaPackages
, rocmPackages
, config
, gpuBackend ? (
if config.cudaSupport
then "cuda"
else if config.rocmSupport
then "rocm"
else "none"
)
}:
assert builtins.elem gpuBackend [ "none" "cuda" "rocm" ];

View File

@ -6,9 +6,16 @@
, mpi
, gfortran
, llvmPackages
, gpuBackend ? "none"
, cudaPackages
, rocmPackages
, config
, gpuBackend ? (
if config.cudaSupport
then "cuda"
else if config.rocmSupport
then "rocm"
else "none"
)
}:
assert builtins.elem gpuBackend [ "none" "cuda" "rocm" ];

View File

@ -6,9 +6,16 @@
, blas
, gfortran
, llvmPackages
, gpuBackend ? "none"
, cudaPackages
, rocmPackages
, config
, gpuBackend ? (
if config.cudaSupport
then "cuda"
else if config.rocmSupport
then "rocm"
else "none"
)
}:
assert builtins.elem gpuBackend [ "none" "cuda" "rocm" ];