How do I perform muliple FFTs on a stack of data in parallel on a GPU?

Question

Adam 2013-3-18

1
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/67578-how-do-i-perform-muliple-ffts-on-a-stack-of-data-in-parallel-on-a-gpu

Hi, I have a stack of images, and I want to compute the 2d fourier-transform of each image in parallel. For example, if I was not using a GPU, I would be doing something like this:

A = rand(256,256,100);
B = zeros(256, 256, 100);
for j = 1:100
    B(:,:,j) = fftshift(fft2(A(:,:,j)));
end;

I would like to do each iteration of this for-loop in parallel. Is there a way to do that using gpuArray and arrayfun, or do I have to define my own cuda kernel?

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

请先登录，再进行评论。

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

Answer 1

Edric Ellis 2013-3-18

1
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/67578-how-do-i-perform-muliple-ffts-on-a-stack-of-data-in-parallel-on-a-gpu#answer_79027

在 MATLAB Online 中打开

FFT2 in MATLAB already applies to each 'page' of a 3-dimensional array, and this is true on the GPU too. Unfortunately, FFTSHIFT doesn't work that way, so you need something like this:

A = gpuArray.rand(256, 256, 100);
B = fftshift(fftshift(fft2(A), 1), 2);

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

Adam 2013-3-18

在 MATLAB Online 中打开

Thanks for your help! This definitely makes things very simple! Out of curiosity, is there a standard way of independently processing multiple independent 'pages' simultaneously by some arbitrary function? For example, if I have a function:

b = myfun(a)

where b and a are both n-by-n, how would I use arrayfun to apply 'myfun' to each page of a n-by-n-by-m gpuArray, (thus returning another 3D gpuArray)?

请先登录，再进行评论。