odeinjul / pytorch-extension-demo

pytorch-extension-demo

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

python3 tests/test_add.py

Language:C++ 90.8%Language:CMake 5.3%Language:Cuda 2.1%Language:Python 1.5%Language:C 0.3%