Skip to content

huangtinglin/PyTorch-extension-Convolution

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

Watchers

Forks

Releases

No releases published

Packages

No packages published