diff --git a/pkgs/by-name/sp/spla/package.nix b/pkgs/by-name/sp/spla/package.nix index c6d662c633ec..4a40fed43dce 100644 --- a/pkgs/by-name/sp/spla/package.nix +++ b/pkgs/by-name/sp/spla/package.nix @@ -50,13 +50,17 @@ stdenv.mkDerivation (finalAttrs: { nativeBuildInputs = [ cmake gfortran - ]; + ] + ++ lib.optionals (gpuBackend == "cuda") [ cudaPackages.cuda_nvcc ]; buildInputs = [ blas mpi ] - ++ lib.optional (gpuBackend == "cuda") cudaPackages.cudatoolkit + ++ lib.optionals (gpuBackend == "cuda") [ + cudaPackages.cuda_cudart + cudaPackages.libcublas + ] ++ lib.optionals (gpuBackend == "rocm") [ rocmPackages.clr rocmPackages.rocblas