This repository contains learning exercises for writing cpp extensions in pytorch.
Reference Pytorch Tutorial and corresponding codebase
This repo contains python, cpp and cuda code.
For python install a conda environment with pytorch
In branch v1_nn_module
In branch v2_cpp_extension
In branch v3_cuda_extension