GPU Tridiagonal Solver(CUDA):非基座2三对角系统

时间:2014-08-10 16:30:25

标签: cuda gpu

是否有任何非基础2矩阵的三角形解算器(例如矩阵如此:500X500)? https://code.google.com/p/tridiagonalsolvers/source/browse/#svn%2Ftrunk%2Ftridiagonalsolvers处的算法仅处理基数2。 对非基2矩阵实现这样的求解器有困难吗?

1 个答案:

答案 0 :(得分:2)

cuSparse库可以覆盖任意长度的三对角系统。

您要查看的两个功能是:

  1. cusparsegtsv():http://docs.nvidia.com/cuda/cusparse/#cusparse-lt-t-gt-gtsv
  2. cusparsegtsv_nopivot():http://docs.nvidia.com/cuda/cusparse/#cusparse-lt-t-gt-gtsv_nopivot
  3. 还有一个批量的三角形解算器。

    当矩阵边长是2的幂时,你将获得最佳性能,但是对于非2次幂矩阵,你可能不会失去很多性能,特别是如果它们不是这样的话。略低于2的幂。