arpack: use openblas on darwin; fix build; enable eigen (#367594)
This commit is contained in:
@@ -1,14 +1,16 @@
|
||||
{ lib, stdenv, fetchFromGitHub, cmake
|
||||
{ lib, stdenv, fetchFromGitHub, cmake, ninja
|
||||
, gfortran, blas, lapack, eigen
|
||||
, useMpi ? false
|
||||
, mpi
|
||||
, openssh
|
||||
, igraph
|
||||
, useAccel ? false #use Accelerate framework on darwin
|
||||
}:
|
||||
|
||||
# MPI version can only be built with LP64 interface.
|
||||
# See https://github.com/opencollab/arpack-ng#readme
|
||||
assert useMpi -> !blas.isILP64;
|
||||
assert useAccel -> stdenv.hostPlatform.isDarwin;
|
||||
|
||||
stdenv.mkDerivation rec {
|
||||
pname = "arpack";
|
||||
@@ -21,35 +23,39 @@ stdenv.mkDerivation rec {
|
||||
sha256 = "sha256-HCvapLba8oLqx9I5+KDAU0s/dTmdWOEilS75i4gyfC0=";
|
||||
};
|
||||
|
||||
nativeBuildInputs = [ cmake gfortran ];
|
||||
buildInputs = assert (blas.isILP64 == lapack.isILP64); [
|
||||
blas
|
||||
lapack
|
||||
nativeBuildInputs = [ cmake gfortran ninja ];
|
||||
buildInputs = [
|
||||
eigen
|
||||
] ++ lib.optional useMpi mpi;
|
||||
] ++ lib.optionals (!useAccel) (assert (blas.isILP64 == lapack.isILP64); [
|
||||
blas lapack
|
||||
]) ++ lib.optional useMpi mpi;
|
||||
|
||||
nativeCheckInputs = lib.optional useMpi openssh;
|
||||
|
||||
# a couple tests fail when run in parallel
|
||||
doCheck = true;
|
||||
enableParallelChecking = false;
|
||||
|
||||
env = lib.optionalAttrs useAccel {
|
||||
# Without these flags some tests will fail / segfault when using Accelerate
|
||||
# framework. They were pulled from the CI Workflow
|
||||
# https://github.com/opencollab/arpack-ng/blob/804fa3149a0f773064198a8e883bd021832157ca/.github/workflows/jobs.yml#L184-L192
|
||||
FFLAGS = "-ff2c -fno-second-underscore";
|
||||
};
|
||||
|
||||
cmakeFlags = [
|
||||
"-DBUILD_SHARED_LIBS=ON"
|
||||
"-DINTERFACE64=${if blas.isILP64 then "1" else "0"}"
|
||||
"-DMPI=${if useMpi then "ON" else "OFF"}"
|
||||
"-DICB=ON"
|
||||
(lib.cmakeBool "BUILD_SHARED_LIBS" stdenv.hostPlatform.hasSharedLibraries)
|
||||
(lib.cmakeBool "EIGEN" true)
|
||||
(lib.cmakeBool "EXAMPLES" true)
|
||||
(lib.cmakeBool "ICB" true)
|
||||
(lib.cmakeBool "INTERFACE64" (!useAccel && blas.isILP64))
|
||||
(lib.cmakeBool "MPI" useMpi)
|
||||
] ++ lib.optionals stdenv.hostPlatform.isDarwin [
|
||||
"-DBLA_VENDOR=${if useAccel then "Apple" else "Generic"}"
|
||||
];
|
||||
|
||||
preCheck = ''
|
||||
# Prevent tests from using all cores
|
||||
export OMP_NUM_THREADS=2
|
||||
'';
|
||||
|
||||
postFixup = lib.optionalString stdenv.hostPlatform.isDarwin ''
|
||||
install_name_tool -change libblas.dylib ${blas}/lib/libblas.dylib $out/lib/libarpack.dylib
|
||||
'';
|
||||
|
||||
passthru = {
|
||||
inherit (blas) isILP64;
|
||||
isILP64 = !useAccel && blas.isILP64;
|
||||
tests = {
|
||||
inherit igraph;
|
||||
};
|
||||
|
||||
Reference in New Issue
Block a user