/cuda_ext_example

Simple example project that builds a PyTorch CUDA extension

Primary LanguageCudaMIT LicenseMIT

Stargazers