gpt4 book ai didi

iPhone GPU 上的大型矩阵乘法

转载 作者:行者123 更新时间:2023-12-03 19:11:50 25 4
gpt4 key购买 nike

我设计了一个用于 iPad 游戏的学习 AI,但其原型(prototype)是用 matlab 编写的。我需要执行多项操作,例如:(大写表示矩阵)

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

我的一些矩阵很大(2601 x 100)。在 GPU 上运行这些操作可以显着提高该操作的性能。有没有比 Accelerate 的 BLAS 更合适的线性代数库? BLAS 最多仅支持 4x4 矩阵。

最佳答案

我们在以前的项目中使用(开源)Eigen 取得了相当好的结果。图书馆。

  • 它是一个仅包含头文件的 C++ 模板库,但这没有问题,因为您可以用纯 C 或 Objective-C++ (.mm) 编写接口(interface)。
  • 其代码针对armv7进行了深度优化。
  • LGPL3 许可模式应该是在应用商店发布应用且不干扰 Apple TOS 的不错选择。

关于iPhone GPU 上的大型矩阵乘法,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/8779580/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com