WIP: shell.nix that provides all the dependencies

libtorch-ffi/libtorch-ffi.cabal

@@ -65,17 +65,17 @@ library
                     , async
  extra-libraries:     stdc++
                     , c10
-                    , iomp5
-                    , mklml
+                    -- , iomp5   -- for now
+                    -- , mklml
                     , torch
  extra-ghci-libraries: stdc++
  if os(darwin)
   ld-options: -Wl,-keep_dwarf_unwind
-  ghc-options:       -optc-D_GLIBCXX_USE_CXX11_ABI=0 -optc-std=c++11 -optc-xc++
+  ghc-options:       -optc-D_GLIBCXX_USE_CXX11_ABI=1 -optc-std=c++11 -optc-xc++
  else
-  ghc-options:       -optc-D_GLIBCXX_USE_CXX11_ABI=0 -optc-std=c++11
- cc-options:        -D_GLIBCXX_USE_CXX11_ABI=0 -std=c++11
- cxx-options:       -D_GLIBCXX_USE_CXX11_ABI=0 -std=c++11
+  ghc-options:       -optc-D_GLIBCXX_USE_CXX11_ABI=1 -optc-std=c++11
+ cc-options:        -D_GLIBCXX_USE_CXX11_ABI=1 -std=c++11
+ cxx-options:       -D_GLIBCXX_USE_CXX11_ABI=1 -std=c++11
  default-extensions:          Strict
                             , StrictData
 

shell.nix

@@ -0,0 +1,58 @@
+{ pkgs ?
+   let
+     # revision: https://github.com/NixOS/nixpkgs/pull/65041
+     rev = "9420c33455c5b3e0876813bdd739bc75b3ccbd8a";
+     url = "https://github.com/NixOS/nixpkgs/archive/${rev}.tar.gz";
+     # nix-prefetch-url --unpack ${url}
+     sha256 = "0l82qfcmip0mfijcxssaliyxayfh0c11bk66yhwna1ixc0s0jjd2";
+     nixpkgs = builtins.fetchTarball { inherit url sha256; };
+   in
+     import nixpkgs { config.allowUnfree = true; config.allowUnsupportedSystem = true; }
+}:
+
+with pkgs;
+
+let
+
+  hsenv = haskellPackages.ghcWithPackages (p: with p; [
+    cabal-install
+    ansi-wl-pprint
+    async
+    bytestring
+    containers
+    exceptions
+    finite-typelits
+    ghc-typelits-knownnat
+    hashable
+    hspec
+    hspec-discover
+    mtl
+    optparse-applicative
+    parsec
+    parsers
+    QuickCheck
+    reflection
+    safe-exceptions
+    sysinfo
+    template-haskell
+    transformers
+    transformers-compat
+    unordered-containers
+    vector
+  ]);
+
+in
+
+stdenv.mkDerivation {
+  name = "hasktorch-dev";
+  buildInputs = [ hsenv mkl python3Packages.pytorchWithoutCuda ];
+  shellHook =
+    let
+      libtorch_path = "${python3Packages.pytorchWithoutCuda}/lib/${python3Packages.python.libPrefix}/site-packages/torch";
+    in
+  ''
+    export CPATH=${libtorch_path}/include/torch/csrc/api/include
+    export LD_LIBRARY_PATH=${libtorch_path}/lib
+  '';
+
+}