search

godweiyang / godweiyang.github.io

Star & fork my blog source code!
https://github.com/godweiyang/hexo-matery-modified
18 stars 12 forks source link

PyTorch自定义CUDA算子教程与运行时间分析 | 韦阳的博客 #235

Open godweiyang opened 3 years ago

godweiyang commented 3 years ago

https://godweiyang.com/2021/03/18/torch-cpp-cuda/

最近因为工作需要，学习了一波CUDA。这里简单记录一下PyTorch自定义CUDA算子的方法，写了一个非常简单的example，再介绍一下正确的PyTorch中CUDA运行时间分析方法。 所有的代码都放在了github上，地址是：https