sahamrit / pytorch_cpp_extension

This repository contains learning exercises for writing cpp extensions in pytorch.

pytorch_cpp_extension

This repository contains learning exercises for writing cpp extensions in pytorch.

Reference Pytorch Tutorial and corresponding codebase

Installation

This repo contains python, cpp and cuda code.

For python install a conda environment with pytorch

Checkpoint Branches

Checkpoint 1 : Use nn Module for custom operator

In branch v1_nn_module

Checkpoint 2 : Use cpp extension for custom operator

In branch v2_cpp_extension

Checkpoint 3 : Use cuda extension for custom operator

In branch v3_cuda_extension