A Simple CUDA Python Extension

Summary

A demo python extension using CUDA

To install as a Python extension

> pip install .

To build and run tests

> python -m venv .venv/
> source .venv/Scripts/activate
> pip install -e requirements.txt
> python tools/build.py

System requirements

1. Python >=3.6
2. C++ compiler toolchain
3. CMake
4. CUDA toolkit (optional)