How and why you want to make your pytorch CUDA/CPP extension with a Makefile
Pytorch cpp extensions provides a good way to augment pytorch with custom functions. The cpp-extension uses the setuptool to compile files. However, as it is mainly used for deployment rather than debugging and development, using the setuptool for development can be slow and cumbersome.
In this repository, I provide an alternative way to compile and debug your custom extension with a makefile.
The associated tutorial can be found at the blog post.
You must have torch installed in your current (virtual environment) python.
git clone https://github.com/chrischoy/MakePytorchPlusPluscd MakePytorchPlusPluspython setup.py install
It automatically selects the maximum number of CPU for parallel compilation.
python example.py