mowangmodi / pytorch-cppcuda-tutorial Public

forked from kwea123/pytorch-cppcuda-tutorial

Notifications You must be signed in to change notification settings
Fork 0
Star 0

tutorial for writing custom pytorch cpp+cuda kernel, applied on volume rendering (NeRF)

0 stars 34 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 10 Commits
.vscode		.vscode
include		include
.gitignore		.gitignore
README.md		README.md
interpolation.cpp		interpolation.cpp
interpolation_kernel.cu		interpolation_kernel.cu
setup.py		setup.py
test.py		test.py

Repository files navigation

pytorch-cppcuda-tutorial

tutorial for writing custom pytorch cpp+cuda kernel, applied on volume rendering (NeRF)

tutorial playlist: https://www.youtube.com/watch?v=l_Rpk6CRJYI&list=PLDV2CyUo4q-LKuiNltBqCKdO9GH4SS_ec&ab_channel=AI%E8%91%B5

CUDA explanation in video 2: https://nyu-cds.github.io/python-gpu/02-cuda/

C++ API in video 3: https://pytorch.org/cppdocs/

kernel launching: https://pytorch.org/tutorials/advanced/cpp_extension.html

About

tutorial for writing custom pytorch cpp+cuda kernel, applied on volume rendering (NeRF)

Report repository

Releases

No releases published

Packages

No packages published

Languages

Cuda 50.2%
Python 35.8%
C++ 8.3%
C 5.7%