Skip to content

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Implemented convolution based on CUDA extensions in PyTorch

A convolution implementation based on cuda extension for PyTorch. The source code reference to the PyTorch's inefficient implementation here. See here for the accompanying tutorial.

  • Build CUDA extensions by going into the conv_cuda/ folder and executing python setup.py install,
  • JIT-compile CUDA extensions by going into the conv_cuda/ folder and calling python jit.py, which will JIT-compile the extension and load it,
  • Check the result of the convolution by running python test.py

About

An example of C++ extension for PyTorch.

Topics

Resources

Stars

37 stars

Watchers

0 watching

Forks

Releases

Packages

Used by

Contributors

Languages