c++ - Thrust中是否有 boost 计算功能的类比？

Question

我想知道是否有一个模拟来 boost 计算函数 (http://www.boost.org/doc/libs/1_61_0/libs/compute/doc/html/boost_compute/advanced_topics.html#boost_compute.advanced_topics.custom_functions) 变成内核并且可以创建为变成内核的嵌入式内联代码？

Answer 1

因此，您可以使用带有 __device__ 限定符的函数/仿函数。链接中显示了 vector 操作 saxpy 的示例，您可以在其中找到仿函数 saxpy_functor

http://docs.nvidia.com/cuda/thrust/#transformations

类似于boost::compute，您也可以使用 thrust lambda 表达式作为

thrust::transform(X.begin(), X.end(), Y.begin(), Y.begin(), A * _1 + _2);

或@RobertCrovella 所示的更多标准 C++ lambda 表达式。