colin/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

cudaPackages.saxpy: now available pre-11.4 with CUDA Toolkit

Browse Source
This commit is contained in:
Connor Baker 2023-12-14 20:12:20 +00:00
parent c27ade81bf
commit 501a1af970
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -72,10 +72,7 @@ let
# Loose packages
cudatoolkit = final.callPackage ../development/cuda-modules/cudatoolkit {};
# 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 {}
);
saxpy = 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