An example of writing a C++/CUDA extension for PyTorch. See
here for the accompanying tutorial.
This repo demonstrates how to write an example extension_cpp.ops.lltm
custom op that has both custom CPU and CUDA kernels.
To build:
pip install .
To test:
python test/test_extension.py
To benchmark Python vs. C++ vs. CUDA:
python test/benchmark.py