pytorch c++/cuda
文章平均质量分 84
风尘23187
这个作者很懒,什么都没留下…
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
Pytorch+cpp_cuda extension 课程二
既然要用到 cuda 那么,就要 .cu(cuda) 文件,这里并没有并行。而是做一个调用的demo。每个点的内插都是独立的,每个点对于每个特征点来说,也是独立的,因此可在这两个方向上并行。N表示有N个正方体,8表示8个特征点,F表示每个特征点的表示方式,F应该等于3吧。和之前的c++程序相比,有以下变化。N表示有N个正方体,3表示坐标。原创 2022-11-14 18:27:49 · 684 阅读 · 0 评论 -
Pytorch+cpp_cuda extension 课程一
pytorch C++ cuda原创 2022-11-11 19:49:53 · 1232 阅读 · 0 评论 -
自定义 C++ 和 CUDA 扩展
pytorch cuda/c++ 拓展原创 2022-11-11 10:40:58 · 1695 阅读 · 0 评论
分享