PyTorch 源码解读之 ：揭秘 C++/CUDA 算子实现和调用全流程

文章预览

点击上方 “ 小白学视觉 ”，选择加" 星标 "或“ 置顶 ” 重磅干货，第一时间送达 编者荐语   文章结合 Python 与 C++ 各自的优点，在 PyTorch 中加入 C++ / CUDA的扩展，详细解释了C++/CUDA 算子实现和调用全流程，让大家更好地使用工具而不为工具所束缚。 作者丨OpenMMLab@知乎 链接丨https://zhuanlan.zhihu.com/p/348555597 “Python 用户友好却运行效率低”，“C++ 运行效率较高，但实现一个功能代码量会远大于 Python”。 平常学习工作中你是否常听到类似的说法？ 在 Python 大行其道的今天，你是否经常也会面临代码的瓶颈，而为运行加速而烦恼呢？ “我的代码刚跑 10 步，隔壁同学的已经跑完第一个 epoch 了。 ”--这究竟是人性的扭曲还是科学的沦丧？ 荀子有言“君子性非异也，善假于物也”。 本期《 源码解读 》带你走进 “ Pytorch 中 (神秘) 的 C++ / CUDA 扩展 "。 本期 ………………………………