diff options
author | Ben Darwin <bcdarwin@gmail.com> | 2020-01-31 15:30:02 -0500 |
---|---|---|
committer | Ben Darwin <bcdarwin@gmail.com> | 2020-05-09 19:29:47 -0400 |
commit | 1e0912d772257c2731e9424e36464930a7a22604 (patch) | |
tree | b555eb310c72a3a623ca018532f2505175d95208 /pkgs/development/ocaml-modules | |
parent | 24fcef1b9eca0861d4575f649306429ae3279412 (diff) | |
download | nixlib-1e0912d772257c2731e9424e36464930a7a22604.tar nixlib-1e0912d772257c2731e9424e36464930a7a22604.tar.gz nixlib-1e0912d772257c2731e9424e36464930a7a22604.tar.bz2 nixlib-1e0912d772257c2731e9424e36464930a7a22604.tar.lz nixlib-1e0912d772257c2731e9424e36464930a7a22604.tar.xz nixlib-1e0912d772257c2731e9424e36464930a7a22604.tar.zst nixlib-1e0912d772257c2731e9424e36464930a7a22604.zip |
ocamlPackages.torch: init at 0.8
Diffstat (limited to 'pkgs/development/ocaml-modules')
-rw-r--r-- | pkgs/development/ocaml-modules/torch/default.nix | 56 |
1 files changed, 56 insertions, 0 deletions
diff --git a/pkgs/development/ocaml-modules/torch/default.nix b/pkgs/development/ocaml-modules/torch/default.nix new file mode 100644 index 000000000000..457259bb1bbd --- /dev/null +++ b/pkgs/development/ocaml-modules/torch/default.nix @@ -0,0 +1,56 @@ +{ stdenv +, buildDunePackage +, fetchFromGitHub +, cmdliner +, ctypes +, npy +, ocaml-compiler-libs +, ppx_custom_printf +, ppx_expect +, ppx_sexp_conv +, sexplib +, stdio +, pytorch +}: + +buildDunePackage rec { + pname = "torch"; + version = "0.8"; + + owner = "LaurentMazare"; + + minimumOCamlVersion = "4.07"; + + src = fetchFromGitHub { + inherit owner; + repo = "ocaml-${pname}"; + rev = version; + sha256 = "19w31paj24pns2ahk9j9rgpkb5hpcd41kfaarxrlddww5dl6pxvi"; + }; + + propagatedBuildInputs = [ + cmdliner + ctypes + npy + ocaml-compiler-libs + pytorch + pytorch.dev + ppx_custom_printf + ppx_expect + ppx_sexp_conv + sexplib + stdio + ]; + + preBuild = ''export LIBTORCH=${pytorch.dev}/''; + + doCheck = true; + checkPhase = "dune runtest"; + + meta = with stdenv.lib; { + inherit (src.meta) homepage; + description = "Ocaml bindings to Pytorch"; + maintainers = [ maintainers.bcdarwin ]; + license = licenses.asl20; + }; +} |