CudaDemo
January 1, 2023 · View on GitHub
A simple demo to show how to implement custom operators in PyTorch with cuda/c++.
More details can be seen in 知乎: 用Cuda实现PyTorch算子.
Usage
With PyTorch installation, simply run pip install -e .
January 1, 2023 · View on GitHub
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 .