A simple demo to show how to implement custom operators in PyTorch with cuda/c++.
More details can be seen in 知乎: 用Cuda实现PyTorch算子.
With PyTorch installation, simply run pip install -e .
A simple demo to show how to implement custom operators in PyTorch with cuda/c++.
More details can be seen in 知乎: 用Cuda实现PyTorch算子.
With PyTorch installation, simply run pip install -e .