swendel/nixpkgs
1
0
Fork 0
mirror of https://github.com/SebastianWendel/nixpkgs.git synced 2024-09-20 04:19:00 +02:00
Code Issues Packages Projects Releases Wiki Activity

cudaPackages.saxpy: only available from CUDA 11.4 and on

Browse source
This commit is contained in:
Connor Baker 2023-11-30 20:39:40 +00:00
parent 0a7dacf94d
commit aaf735eac9
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
pkgs/top-level/cuda-packages.nix
View file

@ -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