From 33d3f27035e94368e8e0d62db909ed9ba131e0d1 Mon Sep 17 00:00:00 2001
From: Pavol Rusnak <pavol@rusnak.io>
Date: Mon, 21 Jun 2021 18:30:20 +0200
Subject: [PATCH] openmpi: make fortran support optional, disable fortran on
aarch64-darwin
---
.../development/libraries/openmpi/default.nix | 20 +++++++++++--------
1 file changed, 12 insertions(+), 8 deletions(-)
diff --git a/pkgs/development/libraries/openmpi/default.nix b/pkgs/development/libraries/openmpi/default.nix
index 2df08426368d..b264c85a913c 100644
--- a/pkgs/development/libraries/openmpi/default.nix
+++ b/pkgs/development/libraries/openmpi/default.nix
@@ -13,20 +13,21 @@
# Enable libfabric support (necessary for Omnipath networks) on x86_64 linux
, fabricSupport ? stdenv.isLinux && stdenv.isx86_64
+
+# Enable Fortran support
+, fortranSupport ? !(stdenv.isDarwin && stdenv.isAarch64)
}:
assert !cudaSupport || cudatoolkit != null;
let
- version = "4.1.1";
-
cudatoolkit_joined = symlinkJoin {
name = "${cudatoolkit.name}-unsplit";
paths = [ cudatoolkit.out cudatoolkit.lib ];
};
in stdenv.mkDerivation rec {
pname = "openmpi";
- inherit version;
+ version = "4.1.1";
src = with lib.versions; fetchurl {
url = "https://www.open-mpi.org/software/ompi/v${major version}.${minor version}/downloads/${pname}-${version}.tar.bz2";
@@ -44,17 +45,19 @@ in stdenv.mkDerivation rec {
find -name "Makefile.in" -exec sed -i "s/\`date\`/$ts/" \{} \;
'';
- buildInputs = with stdenv; [ gfortran zlib ]
- ++ lib.optionals isLinux [ libnl numactl pmix ucx ]
+ buildInputs = [ zlib ]
+ ++ lib.optionals fortranSupport [ gfortran ]
+ ++ lib.optionals stdenv.isLinux [ libnl numactl pmix ucx ]
++ lib.optionals cudaSupport [ cudatoolkit ]
++ [ libevent hwloc ]
- ++ lib.optional (isLinux || isFreeBSD) rdma-core
+ ++ lib.optional (stdenv.isLinux || stdenv.isFreeBSD) rdma-core
++ lib.optional fabricSupport [ libpsm2 libfabric ];
nativeBuildInputs = [ perl ];
- configureFlags = with stdenv; lib.optional (!cudaSupport) "--disable-mca-dso"
- ++ lib.optionals isLinux [
+ configureFlags = lib.optional (!cudaSupport) "--disable-mca-dso"
+ ++ lib.optional (!fortranSupport) "--disable-mpi-fortran"
+ ++ lib.optionals stdenv.isLinux [
"--with-libnl=${libnl.dev}"
"--with-pmix=${pmix}"
"--with-pmix-libdir=${pmix}/lib"
@@ -86,6 +89,7 @@ in stdenv.mkDerivation rec {
sed -i 's:compiler=.*:compiler=${targetPackages.stdenv.cc}/bin/${targetPackages.stdenv.cc.targetPrefix}c++:' \
$out/share/openmpi/mpic++-wrapper-data.txt
+ '' + lib.optionalString fortranSupport ''
sed -i 's:compiler=.*:compiler=${gfortran}/bin/${gfortran.targetPrefix}gfortran:' \
$out/share/openmpi/mpifort-wrapper-data.txt