diff --git a/overlays/default.nix b/overlays/default.nix index 9bf0323..cea48ba 100644 --- a/overlays/default.nix +++ b/overlays/default.nix @@ -20,4 +20,7 @@ _final: prev: { git-repo-go = prev.pkgs.callPackage ../pkgs/git-repo-go { }; + peakperf = prev.pkgs.callPackage ../pkgs/peakperf { + enableCuda = false; # peakperf needs cuda-samples, which is not available in current CUDA. CUDA 11 is broken due to CVEs in freeimage + }; } diff --git a/pkgs/peakperf/default.nix b/pkgs/peakperf/default.nix index ed837ea..22b5eaf 100644 --- a/pkgs/peakperf/default.nix +++ b/pkgs/peakperf/default.nix @@ -24,12 +24,18 @@ effectiveStdenv.mkDerivation (finalAttrs: { buildInputs = [ ] ++ lib.optionals enableCuda [ - cudaPackages.cudart + cudaPackages.cuda_cudart + cudaPackages.cudatoolkit + cudaPackages.cuda-samples ]; - nativeBuildInputs = [ - cmake - ]; + nativeBuildInputs = + [ + cmake + ] + ++ lib.optionals enableCuda [ + cudaPackages.cuda_nvcc + ]; meta = { description = "Achieve peak performance on x86 CPUs and NVIDIA GPUs";