How to use Sum and Dot function in GPU computation with arrayfun?

Question

Viraj Gandhi 2022-12-28

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1885207-how-to-use-sum-and-dot-function-in-gpu-computation-with-arrayfun

评论： Joss Knight 2022-12-29

I am trying to find a dot product of two arrays. The given code below is an example. I am new to GPU computing and any help is appreciated.

The main function is:

a = [1,2,3,4];
b = [5,6,7,8];
a=gpuArray(a);
b=gpuArray(b);
funOut = arrayfun(@protoDotProd,a,b);

The function where the dot product is below:

function funOut = protoDotProd(a,b)
cTemp = a+b; % Just an example. In reality, lots of operations
%% option1
% funOut = dot(a,cTemp);
%% option2
% funOutTemp = a.*cTemp;
% funOut = sum(funOutTemp);
%% option3
funOut = sum(a .* cTemp);

None of these options are working and giving me an error:

Error using dotTest
Function passed as first input argument contains unsupported or unknown function 'sum'  (or 'dot' for option 1).
For more information see Tips.
Error in 'protoDotProd' (line: 6)

1 个评论
显示 -1更早的评论隐藏 -1更早的评论

Voss 2022-12-28

https://www.mathworks.com/matlabcentral/answers/281288-manipulating-arrays-within-gpu-arrayfun#answer_219870

请先登录，再进行评论。

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

Answer 1

Joss Knight 2022-12-28

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1885207-how-to-use-sum-and-dot-function-in-gpu-computation-with-arrayfun#answer_1137572

You cannot perform vector operations in GPU arrayfun. The a and b arguments to your function are not the whole array, they are the scalar values of each element.

Do your element-wise operations ("lots of operations" in your words, followed by the element-wise multiplication) inside your arrayfun function and then do the sum.

Consult the documentation and examples for more info.

2 个评论
显示无隐藏无

Viraj Gandhi 2022-12-29

在 MATLAB Online 中打开

I see your point here in "The a and b arguments to your function are not the whole array, they are the scalar values of each element". arrayfun(@protoDotProd,a,b) is almost similar to write

for i=1:n
    funOut(i) = protoDotProd(a(i),b(i));
end

Am I correct?

Thanks for your help.

Joss Knight 2022-12-29

Exactly correct, yes. Except that you can only perform scalar operations on those scalars (so you can't form an array by concatenating the two values for instance).

请先登录，再进行评论。