diff options
| author | Markus Kowalewski <markus.kowalewski@fysik.su.se> | 2021-11-29 11:31:34 +0100 |
|---|---|---|
| committer | github-actions[bot] <github-actions[bot]@users.noreply.github.com> | 2021-11-29 13:43:14 +0000 |
| commit | 1131ea96cf8db685349cb61813ffe979dbaf7ee0 (patch) | |
| tree | e3abfc69f292eccdddb95799c0c42a4af2259e07 | |
| parent | obelisk: 0.5.2 → 0.6.0 (diff) | |
| download | nixpkgs-1131ea96cf8db685349cb61813ffe979dbaf7ee0.tar.gz | |
ucx: add optional Cuda support
(cherry picked from commit 24fb8db66d32105d05f82f154e4820d079026313)
| -rw-r--r-- | pkgs/development/libraries/ucx/default.nix | 24 |
1 files changed, 20 insertions, 4 deletions
diff --git a/pkgs/development/libraries/ucx/default.nix b/pkgs/development/libraries/ucx/default.nix index 0bb3fe135d66..06c0ada16feb 100644 --- a/pkgs/development/libraries/ucx/default.nix +++ b/pkgs/development/libraries/ucx/default.nix @@ -1,8 +1,17 @@ { lib, stdenv, fetchFromGitHub, autoreconfHook, doxygen -, numactl, rdma-core, libbfd, libiberty, perl, zlib +, numactl, rdma-core, libbfd, libiberty, perl, zlib, symlinkJoin +, enableCuda ? false +, cudatoolkit }: -stdenv.mkDerivation rec { +let + # Needed for configure to find all libraries + cudatoolkit' = symlinkJoin { + inherit (cudatoolkit) name meta; + paths = [ cudatoolkit cudatoolkit.lib ]; + }; + +in stdenv.mkDerivation rec { pname = "ucx"; version = "1.11.2"; @@ -15,7 +24,14 @@ stdenv.mkDerivation rec { nativeBuildInputs = [ autoreconfHook doxygen ]; - buildInputs = [ numactl rdma-core libbfd libiberty perl zlib ]; + buildInputs = [ + libbfd + libiberty + numactl + perl + rdma-core + zlib + ] ++ lib.optional enableCuda cudatoolkit; configureFlags = [ "--with-rdmacm=${rdma-core}" @@ -23,7 +39,7 @@ stdenv.mkDerivation rec { "--with-rc" "--with-dm" "--with-verbs=${rdma-core}" - ]; + ] ++ lib.optional enableCuda "--with-cuda=${cudatoolkit'}"; enableParallelBuilding = true; |