64 lines
1.5 KiB
Nix
64 lines
1.5 KiB
Nix
{ lib, stdenv, fetchFromGitHub, fetchpatch, autoreconfHook, doxygen
|
|
, numactl, rdma-core, libbfd, libiberty, perl, zlib, symlinkJoin
|
|
, enableCuda ? false
|
|
, cudatoolkit
|
|
}:
|
|
|
|
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.13.1";
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "openucx";
|
|
repo = "ucx";
|
|
rev = "v${version}";
|
|
sha256 = "sha256-NhtN8xrHc6UnUrMbq9LHpb25JO+/LDGcLLGebCfGnv4=";
|
|
};
|
|
|
|
patches = [
|
|
# Pull upstream fix for binutils-2.39:
|
|
# https://github.com/openucx/ucx/pull/8450
|
|
(fetchpatch {
|
|
name = "binutils-2.39.patch";
|
|
url = "https://github.com/openucx/ucx/commit/6b6128efd416831cec3a1820f7d1c8e648b79448.patch";
|
|
sha256 = "sha256-ci00nZG8iOUEFXbmgr/5XkIfiw4eAAdG1wcEYjQSiT8=";
|
|
})
|
|
];
|
|
|
|
nativeBuildInputs = [ autoreconfHook doxygen ];
|
|
|
|
buildInputs = [
|
|
libbfd
|
|
libiberty
|
|
numactl
|
|
perl
|
|
rdma-core
|
|
zlib
|
|
] ++ lib.optional enableCuda cudatoolkit;
|
|
|
|
configureFlags = [
|
|
"--with-rdmacm=${rdma-core}"
|
|
"--with-dc"
|
|
"--with-rc"
|
|
"--with-dm"
|
|
"--with-verbs=${rdma-core}"
|
|
] ++ lib.optional enableCuda "--with-cuda=${cudatoolkit'}";
|
|
|
|
enableParallelBuilding = true;
|
|
|
|
meta = with lib; {
|
|
description = "Unified Communication X library";
|
|
homepage = "https://www.openucx.org";
|
|
license = licenses.bsd3;
|
|
platforms = platforms.linux;
|
|
maintainers = [ maintainers.markuskowa ];
|
|
};
|
|
}
|