1
0
Fork 1
mirror of https://github.com/NixOS/nixpkgs.git synced 2024-09-11 15:08:33 +01:00

openmpi: build with ucx support

This commit is contained in:
Markus Kowalewski 2020-07-16 17:34:07 +02:00
parent 0e2f98d69d
commit fe2c68a791
No known key found for this signature in database
GPG key ID: D865C8A91D7025EB

View file

@ -1,6 +1,6 @@
{ stdenv, fetchurl, fetchpatch, gfortran, perl, libnl
, rdma-core, zlib, numactl, libevent, hwloc, targetPackages, symlinkJoin
, libpsm2, libfabric, pmix
, libpsm2, libfabric, pmix, ucx
# Enable CUDA support
, cudaSupport ? false, cudatoolkit ? null
@ -46,7 +46,7 @@ in stdenv.mkDerivation rec {
'';
buildInputs = with stdenv; [ gfortran zlib ]
++ lib.optionals isLinux [ libnl numactl pmix ]
++ lib.optionals isLinux [ libnl numactl pmix ucx ]
++ lib.optionals cudaSupport [ cudatoolkit ]
++ [ libevent hwloc ]
++ lib.optional (isLinux || isFreeBSD) rdma-core