自学内容网 自学内容网

【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)!