Passing a Global Variable to arrayfun for gpuArray

7 visualizaciones (últimos 30 días)
FARHAN ISLAM
FARHAN ISLAM el 3 de Nov. de 2021
Comentada: Jens Risbo el 17 de Feb. de 2022
Hi,
I would like to pass a global array to the arrayfun function. While it does work with regular CPU array, it does not work with GPU array. My original code is quite complicated, but I am pasting a simplified code to help you understand what I am trying to do.
Any help on how this can be accomplished is greatly appreciated!
M = randi(500,100,100);
global a
a = [1 2 3];
tic
C = arrayfun(@function_sum,M);
CPUTime = toc
M = gpuArray(M);
tic
C = arrayfun(@function_sum,M);
GPUTime = toc;
Boost = CPUTime/GPUTime
function y = function_sum(x)
global a
y = 0;
for i = 1 : x
y = y + 1/i + 1*a(1) + 2*a(2) + 3*a(3);
end
end

Respuesta aceptada

Edric Ellis
Edric Ellis el 4 de Nov. de 2021
You can do this by using a nested function and variables in the containing parent workspace instead of global variables. Here's how you would apply this to your example. Note that I've made function_sum become a nested function, and it accesses a directly from the containing function workspace.
function repro
M = randi(500,100,100);
a = [1 2 3];
% Nested function
function y = function_sum(x)
y = 0;
for i = 1 : x
% Access 'a' directly from parent workspace
y = y + 1/i + 1*a(1) + 2*a(2) + 3*a(3);
end
end
tic
C = arrayfun(@function_sum,M);
CPUTime = toc
M = gpuArray(M);
tic
C = arrayfun(@function_sum,M);
GPUTime = toc;
Boost = CPUTime/GPUTime
end
There's a more detailed example of this approach here in the doc.
  2 comentarios
FARHAN ISLAM
FARHAN ISLAM el 5 de Nov. de 2021
Thank you so much Edric! Your method worked and it lowered my calcualtion time on the original code that I am working with by a factor of 60.
Jens Risbo
Jens Risbo el 17 de Feb. de 2022
Thanks for the example ! Just one Question.
Are these varables initialed everytime the nested function is called, or only once?
/Jens

Iniciar sesión para comentar.

Más respuestas (0)

Categorías

Más información sobre GPU Computing en Help Center y File Exchange.

Productos


Versión

R2021b

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by