# PyTorch-extension-Convolution

**Repository Path**: frontxiang/PyTorch-extension-Convolution

## Basic Information

- **Project Name**: PyTorch-extension-Convolution
- **Description**: https://github.com/huangtinglin/PyTorch-extension-Convolution.git
- **Primary Language**: Python
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2023-10-02
- **Last Updated**: 2023-10-02

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# 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](https://github.com/pytorch/pytorch/blob/master/aten/src/THCUNN/generic/SpatialConvolutionMM.cu). See [here](http://pytorch.org/tutorials/advanced/cpp_extension.html) 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`