diff options
| author | hyphon81 <zero812n@gmail.com> | 2017-12-28 17:40:28 +0100 |
|---|---|---|
| committer | Jan Tojnar <jtojnar@gmail.com> | 2017-12-28 17:46:52 +0100 |
| commit | 05573d3e06750c5e1154f72d2e7d96d4e87d2237 (patch) | |
| tree | 038f3bfa4ca530ebe80d0ced8d033327fbe003dc /pkgs | |
| parent | 50a6403b094355bb43d2347dd0c7ea08ee561bb2 (diff) | |
| download | nixlib-05573d3e06750c5e1154f72d2e7d96d4e87d2237.tar nixlib-05573d3e06750c5e1154f72d2e7d96d4e87d2237.tar.gz nixlib-05573d3e06750c5e1154f72d2e7d96d4e87d2237.tar.bz2 nixlib-05573d3e06750c5e1154f72d2e7d96d4e87d2237.tar.lz nixlib-05573d3e06750c5e1154f72d2e7d96d4e87d2237.tar.xz nixlib-05573d3e06750c5e1154f72d2e7d96d4e87d2237.tar.zst nixlib-05573d3e06750c5e1154f72d2e7d96d4e87d2237.zip | |
nccl: init at 1.3.4-1
Diffstat (limited to 'pkgs')
| -rw-r--r-- | pkgs/development/libraries/science/math/nccl/default.nix | 41 | ||||
| -rw-r--r-- | pkgs/top-level/all-packages.nix | 4 |
2 files changed, 45 insertions, 0 deletions
diff --git a/pkgs/development/libraries/science/math/nccl/default.nix b/pkgs/development/libraries/science/math/nccl/default.nix new file mode 100644 index 000000000000..c4deccff5b7e --- /dev/null +++ b/pkgs/development/libraries/science/math/nccl/default.nix @@ -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 ]; + }; +} diff --git a/pkgs/top-level/all-packages.nix b/pkgs/top-level/all-packages.nix index 912714a45509..ac97b71d91bf 100644 --- a/pkgs/top-level/all-packages.nix +++ b/pkgs/top-level/all-packages.nix @@ -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 { }; |