在 Thrust 中评估由元素矩阵运算组成的表达式

Question

我想使用 Thrust 来评估由元素矩阵运算组成的表达式。为了清楚起见，让我们考虑如下表达式：

D=A*B+3*sin(C)

其中A、B、C 和D 是矩阵，当然大小相同。

Thrust Quick Start Guide 提供了saxpy 示例，其中y 既用作输入又用作输出，而在我的情况下，输出参数与输入参数不同，顺便说一下，输入参数不止两个.在Element-by-element vector multiplication with CUDA，考虑了输出与输入不同但只有两个输入的情况。

有人可以就如何使用 Thrust 实现上述表达式（输出矩阵不同于输入和两个以上的输入）提供一些建议（可能还有背后的理由）吗？谢谢。

Answer 1

以下是使用Newton 实现该计算的方法，这是 talonmies 评论中提到的库：

#include <newton/newton.hpp>

int main()
{
  float a[4] = {1.0, 1.0, 1.0, 1.0};
  float b[4] = {2.0, 2.0, 2.0, 2.0};
  float c[4] = {3.0, 3.0, 3.0, 3.0};
  float d[4] = {4.0, 4.0, 4.0, 4.0};

  newton::numeric_vector<float> A = a;
  newton::numeric_vector<float> B = b;
  newton::numeric_vector<float> C = c;
  newton::numeric_vector<float> D = d;

  D = A * B + 3.f * sin(C);

  return 0;
}

该库是使用thrust::zip_iterator 和thrust::transform_iterator 构建的，用于实现具有任意数量输入的表达式。具体可以参考实现。