From 03dd3f66fb1e773fa21ac7145ab9f27f0b6e09e1 Mon Sep 17 00:00:00 2001 From: Samuel Ainsworth Date: Fri, 1 Apr 2022 08:01:09 +0000 Subject: [PATCH] nccl: expose cudatoolkit in passthru --- pkgs/development/libraries/science/math/nccl/default.nix | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/pkgs/development/libraries/science/math/nccl/default.nix b/pkgs/development/libraries/science/math/nccl/default.nix index 5036ad890799..c08b5aa5851d 100644 --- a/pkgs/development/libraries/science/math/nccl/default.nix +++ b/pkgs/development/libraries/science/math/nccl/default.nix @@ -38,6 +38,10 @@ stdenv.mkDerivation rec { enableParallelBuilding = true; + passthru = { + inherit cudatoolkit; + }; + meta = with lib; { description = "Multi-GPU and multi-node collective communication primitives for NVIDIA GPUs"; homepage = "https://developer.nvidia.com/nccl";