可以在GPU上的浮点数组上进行数学运算的库吗？

Question

我想在GPU上的大型浮点数组上进行基本的数学运算(加法,减法,除法,乘法),C++中是否有可以实现此目的的库？

例如,在伪代码中:

A = [1,2,3,...]
B = [2,3,9,...]
C = A+B //[3,5,12,...]
D = A-B //[-1,-1,-6,...]
E = A/B //[0.5,0.6,0.3,...]
F = A*B //[2,6,27,...]

Answer 1

看看Boost.Compute库.它是一个类似C++ STL的库,允许您在GPU(或任何OpenCL兼容设备)上执行许多操作.与Thrust不同,它不仅限于NVIDIA GPU.

源代码在这里:https://github.com/boostorg/compute