nordic-dev.net/nixpkgs
nordic-dev.net/nixpkgs
2
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2024-11-23 23:43:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

nccl: init at 1.3.4-1

Browse Source
This commit is contained in:
hyphon81 2017-12-28 17:40:28 +01:00 committed by Jan Tojnar
parent 50a6403b09
commit 05573d3e06
No known key found for this signature in database
GPG Key ID: 7FAB2A15F7A607A4
2 changed files with 45 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
pkgs/development/libraries/science/math/nccl/default.nix Normal file
View File

@ -0,0 +1,41 @@
{ stdenv, fetchFromGitHub
, gcc5, eject, cudatoolkit
}:
stdenv.mkDerivation rec {
name = "cudatoolkit-${cudatoolkit.majorVersion}-nccl-${version}";
version = "1.3.4-1";
src = fetchFromGitHub {
owner = "NVIDIA";
repo = "nccl";
rev = "v${version}";
sha256 = "0fvnrfn572lc6i2a3xyhbifm53ivcrr46z6cqr3b0bwb1iq79m7q";
};
nativeBuildInputs = [
gcc5
eject
];
propagatedBuildInputs = [
cudatoolkit
];
makeFlags = [
"PREFIX=$(out)"
"CUDA_HOME=${cudatoolkit}"
"CUDA_LIB=${cudatoolkit.lib}/lib"
];
meta = with stdenv.lib; {
description = ''
NVIDIA Collective Communications Library.
Multi-GPU and multi-node collective communication primitives.
'';
homepage = https://developer.nvidia.com/nccl;
license = licenses.bsd3;
platforms = [ "x86_64-linux" ];
maintainers = with maintainers; [ hyphon81 ];
};
}

4
pkgs/top-level/all-packages.nix
View File

@ -3545,6 +3545,10 @@ with pkgs;
nbd = callPackage ../tools/networking/nbd { };
nccl = callPackage ../development/libraries/science/math/nccl {
cudatoolkit = cudatoolkit8;
};
ndjbdns = callPackage ../tools/networking/ndjbdns { };
ndppd = callPackage ../applications/networking/ndppd { };