/
flake.nix
88 lines (71 loc) · 2.29 KB
/
flake.nix
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
{
description = "Environment providing the PETSC and various solvers";
inputs.flake-utils.url = "github:numtide/flake-utils";
inputs.nixpkgs.url = "github:nixos/nixpkgs";
inputs.petsc = {
url = "git+https://gitlab.com/petsc/petsc?tag=v3.18.3";
flake = false;
};
inputs.slepc = {
url = "git+https://gitlab.com/slepc/slepc?tag=v3.18.1";
flake = false;
};
inputs.sowing = {
url = "git+https://bitbucket.com/petsc/pkg-sowing/v1.1.26-p6";
flake = false;
};
inputs.hypre = {
url = "github:hypre-space/hypre";
flake = false;
};
inputs.mumps = {
url = "git+https://bitbucket.com/petsc/pkg-mumps/v5.4.1-p1";
flake = false;
};
outputs = inputs@{ self, nixpkgs, flake-utils, ... }:
flake-utils.lib.eachDefaultSystem (system:
let pkgs = nixpkgs.legacyPackages.${system};
in {
packages =
let
mkPackages = mpi: rec {
inherit mpi;
blas = pkgs.openblasCompat;
mumps = pkgs.callPackage ./mumps {
src = inputs.mumps;
inherit mpi blas scalapack;
};
mumps-serial = pkgs.callPackage ./mumps {
src = inputs.mumps;
mpi = null;
inherit blas scalapack;
};
slepc = pkgs.callPackage ./slepc {
src = inputs.slepc;
inherit mpi petsc blas;
};
sowing = pkgs.callPackage ./sowing {
src = inputs.sowing;
};
scalapack = pkgs.scalapack.override { inherit mpi; };
hypre = pkgs.callPackage ./hypre {
src = inputs.hypre;
inherit mpi;
};
petsc = pkgs.python3Packages.toPythonModule (pkgs.callPackage ./petsc {
src = inputs.petsc;
version = "3.18.3";
inherit mumps sowing mpi blas;
hypre = if mpi == null then null else hypre;
inherit (pkgs.python3Packages) numpy cython;
});
};
in rec {
openmpi = mkPackages pkgs.openmpi;
mpich = mkPackages pkgs.mpich;
serial = mkPackages null;
default = serial;
};
}
);
}