Note meson-build
This example specifically illustrates the use of meson to build a torch C++ extension.
To build, use
python -m pip install --no-build-isolation .
Note that the --no-build-isolation is required, because the extension needs to link against
the torch shared objects. When running without --no-build-isolation, it will link against the
temporary torch installation which is removed after the build process is finished.
An example of writing a C++/CUDA extension for PyTorch. See
here for the accompanying tutorial.
This repo demonstrates how to write an example extension_cpp.ops.mymuladd
custom op that has both custom CPU and CUDA kernels.
The examples in this repo work with PyTorch 2.4+.
To build:
pip install .
To test:
python test/test_extension.py
To benchmark Python vs. C++ vs. CUDA:
python test/benchmark.py