public
/
nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #87447 from bcdarwin/ocaml-torch 

ocamlPackages.torch: init at 0.8
This commit is contained in:
Benjamin Hipple 2020-05-25 20:25:54 -04:00 committed by GitHub
parent 032e57dead 1e0912d772
commit 5e22232f8f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 60 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

56
pkgs/development/ocaml-modules/torch/default.nix Normal file
View File

@ -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;
};
}

4
pkgs/top-level/ocaml-packages.nix
View File

@ -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 { };