如何在 iOS 的 Metal API 中使用计算函数进行乘法运算

Question

我是 iOS 中 Metal API 的新手。所以我的问题是，例如，如何使用 Metal 计算函数进行乘法，

假设我们有两个长度为 2048 的 float[] 数组，我们希望将相应的元素相乘，形成另一个 2048 的 float[] 数组

喜欢这个

res[i] = a[i] * b[i];

使用 a[] 和 b[] 一个包含 2048 个浮点数的数组

和 res[] 另一个包含 2048 个浮点数的数组

下一步是执行那个“操作”，2048 次。

谁能帮帮我。

如果可能，我需要在 Objective-c 中执行此操作，但我也可以快速阅读。

提前谢谢你。

Answer 1

您应该从一个工作示例开始，然后根据您的需要对其进行调整。这是一个运行在 Metal 之上的prefix sum implementation。这是一种渲染实现，而不是计算着色器，因此它能够在 A7 芯片上有效运行。

Answer 2

本教程：https://machinethink.net/blog/mps-matrix-multiplication/ 还有这个：https://www.youtube.com/watch?v=lSofOJrFsJ4&ut=

真的帮了我大忙！