【Pytorch】cumsum的CUDA实现源码位置
cumsum是Pytorch中的一个计算前缀和的算子，通过chatgpt定位其源码位置时，发现chatgpt仅能提供旧的位置，也即位于aten/src/ATen/native/cuda/ScanKernels.cu，但笔者发现该文件已经不存在了，通过搜索发现，在2022年已经将ScanKernels.cu分裂为了几个文件以优化编译速度，特此记录。
原文地址：https://blog.csdn.net/weixin_45207619/article/details/140631845
免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！