diff options
author | Benjamin Hipple <bhipple@protonmail.com> | 2020-05-25 20:25:54 -0400 |
---|---|---|
committer | GitHub <noreply@github.com> | 2020-05-25 20:25:54 -0400 |
commit | 5e22232f8f7dbfb4b60dd72208d96c9b39f54627 (patch) | |
tree | b3590426cef982936995b46eb4dc4b6caa0ac424 /pkgs | |
parent | 032e57dead8cae2c7062cd9a41605206b6fe4d6c (diff) | |
parent | 1e0912d772257c2731e9424e36464930a7a22604 (diff) |
Merge pull request #87447 from bcdarwin/ocaml-torch
ocamlPackages.torch: init at 0.8
Diffstat (limited to 'pkgs')
-rw-r--r-- | pkgs/development/ocaml-modules/torch/default.nix | 56 | ||||
-rw-r--r-- | pkgs/top-level/ocaml-packages.nix | 4 |
2 files changed, 60 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 0000000000000..457259bb1bbd3 --- /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; + }; +} diff --git a/pkgs/top-level/ocaml-packages.nix b/pkgs/top-level/ocaml-packages.nix index 6c45b77abb776..74baab7d0d5aa 100644 --- a/pkgs/top-level/ocaml-packages.nix +++ b/pkgs/top-level/ocaml-packages.nix @@ -758,6 +758,10 @@ let tls = callPackage ../development/ocaml-modules/tls { }; + torch = callPackage ../development/ocaml-modules/torch { + inherit (pkgs.python3Packages) pytorch; + }; + type_conv_108_08_00 = callPackage ../development/ocaml-modules/type_conv/108.08.00.nix { }; type_conv_109_60_01 = callPackage ../development/ocaml-modules/type_conv/109.60.01.nix { }; type_conv_112_01_01 = callPackage ../development/ocaml-modules/type_conv/112.01.01.nix { }; |