treewide: make 'gpuBackend' use 'config.(cuda|rocm)Support' (9709e96e) · Commits · nix / nixpkgs

pkgs/applications/science/chemistry/cp2k/default.nix

+9 −2

Original line number	Diff line number	Diff line
		@@ -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" ];

+8 −1

Original line number	Diff line number	Diff line
		@@ -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" ];

+8 −1

Original line number	Diff line number	Diff line
		@@ -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" ];

+8 −1

Original line number	Diff line number	Diff line
		@@ -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" ];