Repository files navigation

An example of writing a C++/CUDA extension for PyTorch. Seehere for the accompanying tutorial.This repo demonstrates how to write an exampleextension_cpp.ops.mymuladdcustom op that has both custom CPU and CUDA kernels.

The examples in this repo work with PyTorch 2.4+.

To build:

pip install --no-build-isolation -e .

To test:

python test/test_extension.py

Peter Goldsborough,Richard Zou