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

Question

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

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

Answer 1

寻找矩阵数据类型的库是对的。我建议看看ArrayFire。

这是包含受支持类型列表 的快速参考页面。以下是您可以使用 is 运行的函数，它们分为数据分析、线性代数、图像和信号处理、稀疏矩阵等类别，以及用于数据索引、排序、归约、可视化和更快的一组常用算法for 循环。

其他库包括 CULA 或 MAGMA（专注于线性代数）、Thrust（针对 1D 运算）和许多小众学术库。

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