mirror of
https://github.com/NixOS/nixpkgs.git
synced 2024-11-24 07:53:19 +00:00
123 lines
3.2 KiB
Nix
123 lines
3.2 KiB
Nix
{
|
|
lib,
|
|
stdenv,
|
|
fetchzip,
|
|
cctools,
|
|
gfortran,
|
|
python3,
|
|
blas,
|
|
lapack,
|
|
mpiSupport ? true,
|
|
mpi, # generic mpi dependency
|
|
mpiCheckPhaseHook,
|
|
openssh, # required for openmpi tests
|
|
petsc-withp4est ? false,
|
|
hdf5-support ? false,
|
|
hdf5,
|
|
metis,
|
|
parmetis,
|
|
withParmetis ? false,
|
|
pkg-config,
|
|
p4est,
|
|
zlib, # propagated by p4est but required by petsc
|
|
petsc-optimized ? false,
|
|
petsc-scalar-type ? "real",
|
|
petsc-precision ? "double",
|
|
}:
|
|
|
|
# This version of PETSc does not support a non-MPI p4est build
|
|
assert petsc-withp4est -> p4est.mpiSupport;
|
|
|
|
stdenv.mkDerivation rec {
|
|
pname = "petsc";
|
|
version = "3.21.3";
|
|
|
|
src = fetchzip {
|
|
url = "https://web.cels.anl.gov/projects/petsc/download/release-snapshots/petsc-${version}.tar.gz";
|
|
hash = "sha256-dxHa8JUJCN4zRIXMCx7gcvbzFH2SPtkJ377ssIevjgU=";
|
|
};
|
|
|
|
strictDeps = true;
|
|
nativeBuildInputs = [
|
|
python3
|
|
gfortran
|
|
pkg-config
|
|
] ++ lib.optional mpiSupport mpi ++ lib.optional (mpiSupport && mpi.pname == "openmpi") openssh;
|
|
buildInputs = [
|
|
blas
|
|
lapack
|
|
] ++ lib.optional hdf5-support hdf5 ++ lib.optional petsc-withp4est p4est ++ lib.optionals withParmetis [ metis parmetis ];
|
|
|
|
prePatch = lib.optionalString stdenv.hostPlatform.isDarwin ''
|
|
substituteInPlace config/install.py \
|
|
--replace /usr/bin/install_name_tool ${cctools}/bin/install_name_tool
|
|
'';
|
|
|
|
configureFlags = [
|
|
"--with-blas=1"
|
|
"--with-lapack=1"
|
|
"--with-scalar-type=${petsc-scalar-type}"
|
|
"--with-precision=${petsc-precision}"
|
|
"--with-mpi=${if mpiSupport then "1" else "0"}"
|
|
] ++ lib.optionals mpiSupport [
|
|
"--CC=mpicc"
|
|
"--with-cxx=mpicxx"
|
|
"--with-fc=mpif90"
|
|
] ++ lib.optionals (mpiSupport && withParmetis) [
|
|
"--with-metis=1"
|
|
"--with-metis-dir=${metis}"
|
|
"--with-parmetis=1"
|
|
"--with-parmetis-dir=${parmetis}"
|
|
] ++ lib.optionals petsc-optimized [
|
|
"--with-debugging=0"
|
|
"COPTFLAGS=-O3"
|
|
"FOPTFLAGS=-O3"
|
|
"CXXOPTFLAGS=-O3"
|
|
"CXXFLAGS=-O3"
|
|
];
|
|
preConfigure = ''
|
|
patchShebangs ./lib/petsc/bin
|
|
'' + lib.optionalString petsc-withp4est ''
|
|
configureFlagsArray+=(
|
|
"--with-p4est=1"
|
|
"--with-zlib-include=${zlib.dev}/include"
|
|
"--with-zlib-lib=-L${zlib}/lib -lz"
|
|
)
|
|
'' + lib.optionalString hdf5-support ''
|
|
configureFlagsArray+=(
|
|
"--with-hdf5=1"
|
|
"--with-hdf5-fortran-bindings=1"
|
|
"--with-hdf5-include=${hdf5.dev}/include"
|
|
"--with-hdf5-lib=-L${hdf5}/lib -lhdf5"
|
|
)
|
|
'';
|
|
|
|
hardeningDisable = lib.optionals (!petsc-optimized) [
|
|
"fortify"
|
|
"fortify3"
|
|
];
|
|
|
|
configureScript = "python ./configure";
|
|
|
|
enableParallelBuilding = true;
|
|
|
|
# This is needed as the checks need to compile and link the test cases with
|
|
# -lpetsc, which is not available in the checkPhase, which is executed before
|
|
# the installPhase. The installCheckPhase comes after the installPhase, so
|
|
# the library is installed and available.
|
|
doInstallCheck = true;
|
|
installCheckTarget = "check_install";
|
|
nativeInstallCheckInputs = [ mpiCheckPhaseHook ];
|
|
|
|
passthru = {
|
|
inherit mpiSupport;
|
|
};
|
|
|
|
meta = with lib; {
|
|
description = "Portable Extensible Toolkit for Scientific computation";
|
|
homepage = "https://petsc.org/release/";
|
|
license = licenses.bsd2;
|
|
maintainers = with maintainers; [ cburstedde ];
|
|
};
|
|
}
|