pkgs/development/libraries/elpa/default.nix

   1 { lib, stdenv, fetchurl, autoreconfHook, gfortran, perl
   2 , mpi, blas, lapack, scalapack, openssh
   3 # CPU optimizations
   4 , avxSupport ? stdenv.hostPlatform.avxSupport
   5 , avx2Support ? stdenv.hostPlatform.avx2Support
   6 , avx512Support ? stdenv.hostPlatform.avx512Support
   7 # Enable NIVIA GPU support
   8 # Note, that this needs to be built on a system with a GPU
   9 # present for the tests to succeed.
  10 , enableCuda ? false
  11 # type of GPU architecture
  12 , nvidiaArch ? "sm_60"
  13 , cudatoolkit
  14 } :
  15
  16 assert blas.isILP64 == lapack.isILP64;
  17 assert blas.isILP64 == scalapack.isILP64;
  18
  19 stdenv.mkDerivation rec {
  20   pname = "elpa";
  21   version = "2022.05.001";
  22
  23   passthru = { inherit (blas) isILP64; };
  24
  25   src = fetchurl {
  26     url = "https://elpa.mpcdf.mpg.de/software/tarball-archive/Releases/${version}/elpa-${version}.tar.gz";
  27     sha256 = "sha256-IH5vJtZTL7cDc6/D7z04JVITr2He9lnCXa06MOT8o4s=";
  28   };
  29
  30   patches = [
  31     # Use a plain name for the pkg-config file
  32     ./pkg-config.patch
  33   ];
  34
  35   postPatch = ''
  36     patchShebangs ./fdep/fortran_dependencies.pl
  37     patchShebangs ./test-driver
  38
  39     # Fix the test script generator
  40     substituteInPlace Makefile.am --replace '#!/bin/bash' '#!${stdenv.shell}'
  41   '';
  42
  43   nativeBuildInputs = [ autoreconfHook perl openssh ];
  44
  45   buildInputs = [ mpi blas lapack scalapack ]
  46     ++ lib.optional enableCuda cudatoolkit;
  47
  48   preConfigure = ''
  49     export FC="mpifort"
  50     export CC="mpicc"
  51     export CXX="mpicxx"
  52     export CPP="cpp"
  53
  54     # These need to be set for configure to succeed
  55     export FCFLAGS="${lib.optionalString stdenv.hostPlatform.isx86_64 "-msse3 "
  56       + lib.optionalString avxSupport "-mavx "
  57       + lib.optionalString avx2Support "-mavx2 -mfma "
  58       + lib.optionalString avx512Support "-mavx512"}"
  59
  60     export CFLAGS=$FCFLAGS
  61   '';
  62
  63   configureFlags = [
  64     "--with-mpi"
  65     "--enable-openmp"
  66     "--without-threading-support-check-during-build"
  67   ] ++ lib.optional blas.isILP64 "--enable-64bit-integer-math-support"
  68     ++ lib.optional (!avxSupport) "--disable-avx"
  69     ++ lib.optional (!avx2Support) "--disable-avx2"
  70     ++ lib.optional (!avx512Support) "--disable-avx512"
  71     ++ lib.optional (!stdenv.hostPlatform.isx86_64) "--disable-sse"
  72     ++ lib.optional stdenv.hostPlatform.isx86_64 "--enable-sse-assembly"
  73     ++ lib.optionals enableCuda [  "--enable-nvidia-gpu" "--with-NVIDIA-GPU-compute-capability=${nvidiaArch}" ];
  74
  75   doCheck = true;
  76
  77   preCheck = ''
  78     #patchShebangs ./
  79
  80     # make sure the test starts even if we have less than 4 cores
  81     export OMPI_MCA_rmaps_base_oversubscribe=1
  82
  83     # Fix to make mpich run in a sandbox
  84     export HYDRA_IFACE=lo
  85
  86     # Run dual threaded
  87     export OMP_NUM_THREADS=2
  88
  89     # Reduce test problem sizes
  90     export TEST_FLAGS="1500 50 16"
  91   '';
  92
  93   meta = with lib; {
  94     description = "Eigenvalue Solvers for Petaflop-Applications";
  95     homepage = "https://elpa.mpcdf.mpg.de/";
  96     license = licenses.lgpl3Only;
  97     platforms = platforms.linux;
  98     maintainers = [ maintainers.markuskowa ];
  99   };
 100 }