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 .
-
Notifications
You must be signed in to change notification settings - Fork 8
YuxueYang1204/CudaDemo
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Implement custom operators in PyTorch with cuda/c++
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published