Run custom function on GPU with Parallel Computing Toolbox?

Question

Subhrajit Debnath 2012-5-18

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/38778-run-custom-function-on-gpu-with-parallel-computing-toolbox

I have a function which needs to find min( b -A*x ). The dimensions of b = 5 X 1; A = 5 X 2; x = 2 X 10; The output of A*x would be 5 X 10. and since b is a column vector, the final output will be 5 X 10, [b - A*x] and if we apply min operator we would get 5 X 1. When I try to use the arrayfun function it gives error :- Error using parallel.gpu.GPUArray/arrayfun Matrix dimensions must agree for 'mtimes'

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

Jill Reese 2012-5-18

2
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/38778-run-custom-function-on-gpu-with-parallel-computing-toolbox#answer_48302

Arrayfun on the GPU only supports elementwise operations, so it is not the command that you want to use in this case.

You need to use min and * (mtimes) directly, which are overloaded to run on the GPU. If at least one of A, b, or x are stored on the GPU (i.e. of type parallel.gpu.GPUArray), then your code snippet will run on the GPU.