Apr, 2022

TorchSparse: 高效点云推理引擎

TL;DR本文介绍 TorchSparse,一种高效的基于 GPU 加速的稀疏卷积计算引擎,应用于 AR/VR、自动驾驶等方面。通过采用自适应矩阵乘法分组技术和集成向量化、量化及融合的本地化优化存储访问技术,实现了 1.4-1.5 倍的速度提升和 2.7 倍的内存移动代价降低,相较当前最优方法 MinkowskiEngine 和 SpConv,分别实现了 1.6 倍和 1.5 倍的整体加速效果。