From 938af386b02ba3a514878292ab90a73da37e2c28 Mon Sep 17 00:00:00 2001 From: Markus Kowalewski Date: Tue, 11 Oct 2022 18:30:05 +0200 Subject: [PATCH] ucc: init at 1.1.0 --- pkgs/development/libraries/ucc/default.nix | 50 ++++++++++++++++++++++ pkgs/top-level/all-packages.nix | 2 + 2 files changed, 52 insertions(+) create mode 100644 pkgs/development/libraries/ucc/default.nix diff --git a/pkgs/development/libraries/ucc/default.nix b/pkgs/development/libraries/ucc/default.nix new file mode 100644 index 000000000000..1e626911f28a --- /dev/null +++ b/pkgs/development/libraries/ucc/default.nix @@ -0,0 +1,50 @@ +{ stdenv, lib, fetchFromGitHub, libtool, automake, autoconf, ucx +, enableCuda ? false +, cudatoolkit +, enableAvx ? stdenv.hostPlatform.avxSupport +, enableSse41 ? stdenv.hostPlatform.sse4_1Support +, enableSse42 ? stdenv.hostPlatform.sse4_2Support +} : + +stdenv.mkDerivation rec { + pname = "ucc"; + version = "1.1.0"; + + src = fetchFromGitHub { + owner = "openucx"; + repo = "ucc"; + rev = "v${version}"; + sha256 = "sha256-5rf08SXy+vCfnz4zLJ0cMnxwso4WpZOt0jRRAUviVFU="; + }; + + enableParallelBuilding = true; + + postPatch = '' + + for comp in $(find src/components -name Makefile.am); do + substituteInPlace $comp \ + --replace "/bin/bash" "${stdenv.shell}" + done + ''; + + preConfigure = '' + ./autogen.sh + ''; + + nativeBuildInputs = [ libtool automake autoconf ]; + buildInputs = [ ucx ] + ++ lib.optional enableCuda cudatoolkit; + + configureFlags = [ ] + ++ lib.optional enableSse41 "--with-sse41" + ++ lib.optional enableSse42 "--with-sse42" + ++ lib.optional enableAvx "--with-avx" + ++ lib.optional enableCuda "--with-cuda=${cudatoolkit}"; + + meta = with lib; { + description = "Collective communication operations API"; + license = licenses.bsd3; + maintainers = [ maintainers.markuskowa ]; + platforms = platforms.linux; + }; +} diff --git a/pkgs/top-level/all-packages.nix b/pkgs/top-level/all-packages.nix index 987f3341dc00..b7236d3cb5b9 100644 --- a/pkgs/top-level/all-packages.nix +++ b/pkgs/top-level/all-packages.nix @@ -10527,6 +10527,8 @@ with pkgs; mpi = openmpi; # this attribute should used to build MPI applications + ucc = callPackage ../development/libraries/ucc {}; + ucx = callPackage ../development/libraries/ucx {}; openmodelica = recurseIntoAttrs (callPackage ../applications/science/misc/openmodelica {});