Add support in nix for mkl

2021-10-13 12:22:58 +02:00
parent b72303cad4
commit a9ddbf4f1a
3 changed files with 52 additions and 20 deletions
--- a/etc/nix/mkl.nix
+++ b/etc/nix/mkl.nix
@@ -0,0 +1,13 @@
+{ pkgs, ...}:
+
+{
+  buildInputs = with pkgs; [
+    mkl
+  ];
+
+  shellHook = ''
+    export MKL_PATH=${pkgs.mkl}
+    export LD_LIBRARY_PATH=$MKL_PATH:$LD_LIBRARY_PATH
+  '';
+}
+
--- a/etc/nix/openblas.nix
+++ b/etc/nix/openblas.nix
@@ -0,0 +1,16 @@
+{ pkgs, ...}:
+
+{
+  buildInputs = with pkgs; [
+    openblas
+    scalapack
+  ];
+
+  shellHook = ''
+    export OPENBLAS_PATH=${pkgs.openblas}
+    export SCALAPACK_PATH=${pkgs.scalapack}
+    export LD_LIBRARY_PATH=${pkgs.scalapack}/lib:$LD_LIBRARY_PATH
+  '';
+}
+
+