parallel-processing - Matlab GPU 计算 : Process matrix as one unit

Question

我有一个名为 fun 的函数实现如下:

function result = fun( number , matrix)
 result = var( mean(matrix) + number ); // any processing not that exactly
end

使用 matlab parallel Computing toolbox我想将它分布在 GPU 上一组具有相同矩阵的数字上，如下面的代码:

mat = magic(10);
numbers = 1:5;
gMat = gpuArray(mat);
gNumbers = gpuArray(numbers);
result = arrayfun(@fun,gNumbers,gMat); // Error here

如果有人可以提供有关此问题的帮助，那就太好了。

Answer 1

在最初提出的情况下，您可以使用 MATLAB 的 BSXFUN在 GPU 上。 (不能以这种方式解决编辑后的问题。)

mat = magic(5);
numbers = 1:5;
gMat = gpuArray(mat);
gNumbers = gpuArray(numbers);
result = bsxfun(@rdivide, gMat, gNumbers);

请注意，我使用了 @rdivide直接而不是调用单独的函数 - 您不需要这样做，但它使事情变得更短。你也可以像这样使用匿名函数:

bsxfun(@(x,y) x./y, gMat, gNumbers)