iPhone上基于GPU的大矩阵乘法

Question

4

我为iPad游戏设计了一个学习型AI，但其原型是用matlab编写的。我需要执行以下几个操作：（大写表示矩阵）

A = B > c;
A = B * C;
A = B' * C;
A = B .* C;
A = B - C;

我有一些大型矩阵（2601 x 100）。在GPU上运行这些操作可以显著提高此操作的性能。是否有一个更适合的线性代数库来替换Accelerate的BLAS？因为BLAS仅支持最大4x4矩阵。

- Alex Londeree

经过几天的搜索，我找到了ViennaCL。它可以在GPU上执行线性代数运算。但我不知道它在iPad上的表现如何，等有机会我会进行一些基准测试。 - Alex Londeree

1个回答

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

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

- Kay · Accepted Answer

我们在以前的项目中使用了（开源）Eigen库，取得了不错的结果。