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/development/ocaml-modules | |
parent | 032e57dead8cae2c7062cd9a41605206b6fe4d6c (diff) | |
parent | 1e0912d772257c2731e9424e36464930a7a22604 (diff) | |
download | nixlib-5e22232f8f7dbfb4b60dd72208d96c9b39f54627.tar nixlib-5e22232f8f7dbfb4b60dd72208d96c9b39f54627.tar.gz nixlib-5e22232f8f7dbfb4b60dd72208d96c9b39f54627.tar.bz2 nixlib-5e22232f8f7dbfb4b60dd72208d96c9b39f54627.tar.lz nixlib-5e22232f8f7dbfb4b60dd72208d96c9b39f54627.tar.xz nixlib-5e22232f8f7dbfb4b60dd72208d96c9b39f54627.tar.zst nixlib-5e22232f8f7dbfb4b60dd72208d96c9b39f54627.zip |
Merge pull request #87447 from bcdarwin/ocaml-torch
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; + }; +} |