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/development/libraries | |
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/development/libraries')
-rw-r--r-- | pkgs/development/libraries/science/math/nccl/default.nix | 41 |
1 files changed, 41 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 ]; + }; +} |