From aaf735eac97989ee924f2b3296d0d5200a6d734c Mon Sep 17 00:00:00 2001 From: Connor Baker Date: Thu, 30 Nov 2023 20:39:40 +0000 Subject: [PATCH] cudaPackages.saxpy: only available from CUDA 11.4 and on --- pkgs/top-level/cuda-packages.nix | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/pkgs/top-level/cuda-packages.nix b/pkgs/top-level/cuda-packages.nix index 5d4d2fcf29247..19ff558afbc7d 100644 --- a/pkgs/top-level/cuda-packages.nix +++ b/pkgs/top-level/cuda-packages.nix @@ -69,7 +69,10 @@ let # Loose packages cudatoolkit = final.callPackage ../development/cuda-modules/cudatoolkit {}; - saxpy = final.callPackage ../development/cuda-modules/saxpy {}; + # SaxPy is only available after 11.4 because it requires redistributable versions of CUDA libraries. + saxpy = attrsets.optionalAttrs (strings.versionAtLeast cudaVersion "11.4") ( + final.callPackage ../development/cuda-modules/saxpy {} + ); } # NCCL is not supported on Jetson, because it does not use NVLink or PCI-e for inter-GPU communication. # https://forums.developer.nvidia.com/t/can-jetson-orin-support-nccl/232845/9