HartreeFockRustProgram/flake.nix at main · jlheflin/HartreeFockRustProgram · GitHub

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
{
  description = "Rust project with BLAS/LAPACK using Nix flakes";

  inputs = {
    nixpkgs.url = "github:NixOS/nixpkgs/nixos-unstable";
    flake-utils.url = "github:numtide/flake-utils";
    rust-overlay.url = "github:oxalica/rust-overlay";
  };

  outputs = { self, nixpkgs, flake-utils, rust-overlay }:
    flake-utils.lib.eachDefaultSystem (system:
      let
        pkgs = import nixpkgs {
          inherit system;
          overlays = [ rust-overlay.overlays.default ];
        };

        rust = pkgs.rust-bin.stable.latest.default;

      in {
        # ⛏ Build Rust project
        packages.default = pkgs.rustPlatform.buildRustPackage {
          pname = "hf_rust";
          version = "0.1.0";

          src = ./.;

          cargoLock = {
            lockFile = ./Cargo.lock;
          };

          nativeBuildInputs = [
            pkgs.pkg-config
            pkgs.gcc
          ];

          buildInputs = [
            pkgs.openblas
            pkgs.lapack
          ];

          # Optional for linking
          LD_LIBRARY_PATH = "${pkgs.openblas}/lib:${pkgs.lapack}/lib";

          # Optional: helps linking with BLAS in crates like ndarray-linalg
          RUSTFLAGS = "-L${pkgs.openblas}/lib -L${pkgs.lapack}/lib";
        };

        # 🧪 Development shell
        devShell = pkgs.mkShell {
          buildInputs = [
            rust
            pkgs.pkg-config
            pkgs.openblas
            pkgs.lapack
            pkgs.gcc
          ];

          shellHook = ''
            export RUST_BACKTRACE=1
            export LAPACK_LIB=openblas
            export BLAS_LIB=openblas
          '';
        };
      });
}