pytorch/extension-cpp resource

Health v2 · maintenance only

Flags: no_releases no_license

Adoption not part of the score

1203 stars · 247 forks observed · 2026-08-28

What it is AI-extracted, prompt v1, taxonomy v1, 2026-08-30, confidence not recorded

An official PyTorch example repository demonstrating how to write C++/CUDA extensions with custom ops. It includes two variants: one using the standard ATen/LibTorch API and one using the LibTorch Stable ABI.

Use cases

• write custom C++ CUDA kernels for pytorch
• create a custom op in pytorch with cpu and cuda support
• learn how to build pytorch cpp extensions
• build a pytorch extension that works across pytorch versions
• speed up a slow python op with a c++ kernel

When to choose

• you need a reference example for writing pytorch C++/CUDA custom ops
• you want to learn the standard ATen extension API or the new stable ABI
• you need a starting template for a pytorch extension project

When to avoid

• you need a production library rather than example code
• you want prebuilt extensions without writing C++ yourself
• you need extensions for pytorch versions older than 2.10 with the stable ABI variant

Facets

learning-resource · maturity active

machine-learning gpu-computing developer-tools deep-learning developer-tools gpu-computing python cpp cross-platform pytorch cpp-extension cuda custom-ops example-code stable-abi libtorch gpu

Member repositories

For agents

markdown · JSON · MCP: product_card(name="pytorch/extension-cpp")