pytorch-extension-demo

Env

  • pytorch

Build

$ mkdir build
$ cd build
$ cmake .. & make -j

Run

python3 tests/test_add.py