2
海洋气象
海光支持高性能的稀疏卷积吗?
2025-09-24 19:26:07
121
我知道AMD有rocSPARSE和hipSPARSE这两个用于稀疏计算的函数库,但是我想要的是对稀疏3维数据(最典型的像点云数据)的卷积函数,我知道海光有MIOpen这样的深度学习库,但就我所知MIOpen提供的是常规的卷积(稠密数据的卷积)支持,并没有稀疏卷积的支持。请问社区工作人员稀疏卷积有专门的开发吗?在哪里获得相应的函数库?
全部评论(5)
只看楼主
网
网友ehVriI
1楼
你好,当前我们自研的卷积算子库,还不支持这种稀疏卷积的需求。后期我们评估一下,这种需求的重要性,是否合入下一版的开发中。
2025/9/25
网
网友张顺飞
回复 @网友ehVriI
好的,再请问一下,cuda生态下专门用来实现稀疏卷积的Minkowski Engine, spconv, 或 torchsparse这三个库是否有适配或者适配计划?
2025/9/25
C
changhl
回复 @网友张顺飞
您好:torchsparse适配过,基于dtk2504及之后的版本没有适配。 对于AI框架,对外提供了移植方案,您可以看下https://download.sourcefind.cn:65024/4/main/%E6%8A%80%E6%9C%AF%E6%96%87%E6%A1%A3 下的《fastpt使用文档》,如果移植过程中遇到问题,随时反馈。
2025/9/26
网
网友张顺飞
回复 @changhl
你好,感谢回复。你提到torchsparse适配过,请问相关的资源和文档在哪里能够看到或者使用?在DAS中用sparse关键词搜索发现dtk24的Pytorch Sparse ( https://das.sourcefind.cn:55011/portal/#/installation?id=5b03af11-6c18-11ef-bb3e-005056904552&type=frame )但这个只是pytorch的一个拓展不能执行稀疏卷积操作,而我说的torchsparse是指 https://github.com/mit-han-lab/torchsparse/tree/master 想再次确认一下适配情况,如果torchsparse之前有过适配请问在哪里可以找到资源?
2025/9/26
全部4条回复
没有更多内容了
--
楼主
Level.3
0 帖子
楼主热帖
暂无数据