mirror of
https://github.com/ilyakooo0/nixpkgs.git
synced 2024-09-21 04:28:40 +03:00
treewide: make 'gpuBackend' use 'config.(cuda|rocm)Support'
This commit is contained in:
parent
242544ca17
commit
9709e96e1e
@ -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" ];
|
||||
|
@ -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" ];
|
||||
|
@ -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" ];
|
||||
|
@ -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" ];
|
||||
|
Loading…
Reference in New Issue
Block a user