pytorch/extension-cpp
Example C++ and CUDA extensions for PyTorch demonstrating custom op implementation using ATen/LibTorch APIs.
Not currently ranked — collecting fresh signals.
star history
This repository provides two reference implementations of PyTorch custom operations written in C++ and CUDA. The extensions demonstrate both the standard ATen/LibTorch API and the newer Stable ABI approach for writing version-compatible custom ops. Both examples implement a simple mymuladd operation with CPU and CUDA kernels, serving as boilerplate for developers wanting to extend PyTorch’s backend with custom kernels.
Frequently asked
- What is pytorch/extension-cpp?
- Example C++ and CUDA extensions for PyTorch demonstrating custom op implementation using ATen/LibTorch APIs.
- Is extension-cpp open source?
- Yes — pytorch/extension-cpp is an open-source project tracked on heatdrop.
- What language is extension-cpp written in?
- pytorch/extension-cpp is primarily written in Python.
- How popular is extension-cpp?
- pytorch/extension-cpp has 1.2k stars on GitHub.
- Where can I find extension-cpp?
- pytorch/extension-cpp is on GitHub at https://github.com/pytorch/extension-cpp.