https://github.com/YuxueYang1204/CudaDemo

Implement custom operators in PyTorch with cuda/c++
https://github.com/YuxueYang1204/CudaDemo

Last synced: 2 months ago
JSON representation

Implement custom operators in PyTorch with cuda/c++

• Host: GitHub
• URL: https://github.com/YuxueYang1204/CudaDemo
• Owner: YuxueYang1204
• Created: 2023-01-01T08:37:29.000Z (about 2 years ago)
• Default Branch: main
• Last Pushed: 2023-01-01T12:23:41.000Z (about 2 years ago)
• Last Synced: 2024-08-04T02:06:40.309Z (6 months ago)
• Language: Python
• Size: 3.91 KB
• Stars: 23
• Watchers: 2
• Forks: 4
• Open Issues: 1
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

• awesome-cuda-triton-hpc - YuxueYang1204/CudaDemo
• awesome-cuda-triton-hpc - YuxueYang1204/CudaDemo

README

        
# CudaDemo

A simple demo to show how to implement custom operators in PyTorch with cuda/c++.

More details can be seen in [知乎: 用Cuda实现PyTorch算子](https://zhuanlan.zhihu.com/p/595851188).

## Usage

With PyTorch installation, simply run `pip install -e .`

## More

[CUSTOM C++ AND CUDA EXTENSIONS](https://pytorch.org/tutorials/advanced/cpp_extension.html)

[THE C++ FRONTEND](https://pytorch.org/cppdocs/frontend.html)

[LIBRARY API](https://pytorch.org/cppdocs/api/library_root.html)

[详解PyTorch编译并调用自定义CUDA算子的三种方式](https://zhuanlan.zhihu.com/p/358778742)