GPU上稀疏矩阵的基本线性代数
cuSPARSE库为稀疏矩阵提供了GPU加速的基本线性代数子例程,这些子例程的执行速度明显快于仅CPU替代方法。提供了可用于构建GPU加速求解器的功能。cuSPARSE被从事机器学习,计算流体力学,地震勘探和计算科学等应用的工程师和科学家广泛使用。使用cuSPARSE,应用程序会自动受益于常规性能的改进和新的GPU架构。cuSPARSE库包含在NVIDIA HPC SDK和CUDA Toolkit中。
cuSPARSE性能
cuSPARSE库针对NVIDIA GPU的性能进行了高度优化,其SpMM性能比仅CPU的替代产品快30-150倍。
cuSPARSE的主要功能
• 支持密集,COO,CSR,CSC和Blocked CSR稀疏矩阵格式
• 全套稀疏例程,包括稀疏向量x稠密向量运算,稀疏矩阵x稠密向量运算以及稀疏矩阵x稠密矩阵运算。
• 稀疏矩阵x稀疏矩阵加法和乘法的例程
• 稀疏密集向量乘法(SpVV),稀疏矩阵密集向量乘法(SpMV)和稀疏矩阵密集矩阵乘法(SpMM)的通用高性能API
• ILU0和IC0预调节器