cuda-ext-example

Simple example project that builds a PyTorch CUDA extension


Install
pip install cuda-ext-example==0.0.2