jolheiser/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

ucx: add optional Cuda support

Browse Source
This commit is contained in:
Markus Kowalewski 2021-11-29 11:31:34 +01:00
parent a5552c62db
commit 24fb8db66d
No known key found for this signature in database
GPG Key ID: 502A248E3FB4FF48
1 changed files with 20 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
pkgs/development/libraries/ucx/default.nix
View File

@ -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;