CUDA中是否有内置类型的矩阵用于矩阵和矩阵向量运算？

Question

我想实现一些矩阵向量数学。有一些矢量类型：float2，int2，但我在CUDA中找不到任何内置类型矩阵。

是否有适合此类操作的库？

Answer 1

您正确寻找矩阵数据类型的库。我建议您查看ArrayFire。

Here is the quick reference page，其中列出了受支持的类型。 Here are the functions你可以运行is，它被组织成数据分析，线性代数，图像和信号处理，稀疏矩阵以及一系列用于数据索引，排序，缩减，可视化和循环更快。

其他图书馆包括CULA或MAGMA（专注于线性代数），Thrust（针对一维操作）以及一系列利基学术图书馆。

免责声明：我自己在ArrayFire上工作。