andr3/gotch
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
v0.6.2
gotch/gen
History
sugarme 1c2e6f2599 change package 'tensor' to 'ts'
2022-03-12 18:20:20 +11:00
..
pytorch upgrade to libtorch 1.10 2021-11-06 14:44:27 +11:00
dune initial commit 2020-05-22 23:43:09 +10:00
gen.ml change package 'tensor' to 'ts' 2022-03-12 18:20:20 +11:00
gen.ml.1.9 upgrade to libtorch 1.10 2021-11-06 14:44:27 +11:00
gen.mli initial commit 2020-05-22 23:43:09 +10:00
README.md added README.md for generating APIs 2020-11-12 17:25:24 +11:00

README.md

Generate APIs

Get Declaration.yaml file

# master branch can be replaced with release version. E.g., v1.7.0
git clone -b master --recurse-submodule https://github.com/pytorch/pytorch.git
mkdir pytorch-build
cd pytorch-build
cmake -DBUILD_SHARED_LIBS:BOOL=ON -DCMAKE_BUILD_TYPE:STRING=Release -DPYTHON_EXECUTABLE:PATH=`which python3` -DCMAKE_INSTALL_PREFIX:PATH=../pytorch-install ../pytorch
cmake --build . --target install

The Declaration.yaml file is artifact of building Libtorch from source. After running step 4 (can take a while - couple of hours)

cmake -DBUILD_SHARED_LIBS:BOOL=ON -DCMAKE_BUILD_TYPE:STRING=Release -DPYTHON_EXECUTABLE:PATH=`which python3` -DCMAKE_INSTALL_PREFIX:PATH=../pytorch-install ../pytorch

.yaml file can be found in either: pytorch-install/share/ATEN/Declarations.yaml or pytorch-build/aten/src/ATen/Declarations.yaml

Ref.

  1. https://github.com/pytorch/pytorch/blob/master/docs/libtorch.rst
  2. https://discuss.pytorch.org/t/compile-libtorch-c-api-from-source/81624
  3. https://github.com/pytorch/pytorch/issues/12562

Generate APIs

run from root folder (gotch):

dune exec gen/gen.exe