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

Question

3

我想要实现一些矩阵-向量计算。CUDA中有像float2和int2这样的向量类型，但是我找不到任何内置的矩阵类型。

是否有适合这种操作的库？

- jalal sadeghi

翻译：mtrix m; float3 vec;float3 vecOut=m*vec - jalal sadeghi

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- arrayfire · Accepted Answer

你很明智地寻找矩阵数据类型的库。我建议看一下ArrayFire。

这里是快速参考页面，列出了支持的类型。这里是可以运行的函数，它们按数据分析、线性代数、图像和信号处理、稀疏矩阵以及用于数据索引、排序、约简、可视化和更快的循环等常见算法进行分类。

其他库包括CULA或MAGMA（专注于线性代数）、Thrust（针对1D操作）以及一系列小众学术库。

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