for loops vectorization for GPU

Question

Youngjin Jeon 2021-4-27

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/814685-for-loops-vectorization-for-gpu

回答： Ishu 2024-4-4

test_GPU.m

在 MATLAB Online 中打开

hello people,

i want to change for loops for GPU computing.

How to effectively vectorize the loop in this code? i get stuck on this problem. help me plz :)

pnx=101;
pny=101;
pnz=101;
[pY, pX] = meshgrid(linspace(-0.5,0.5,pnx),linspace(-0.5,0.5,pny));
cnt_t=2601;
cnt_r=2601;
tr_x=linspace(-1,1,cnt_t);
tr_y=linspace(-1,1,cnt_t);
re_x=linspace(-2,2,cnt_r);
re_y=linspace(-2,2,cnt_r);
pza=linspace(0,5,101);
Obj=ones(101,101,101);
Obj(:,:,30)=Obj(:,:,30).*2;
Obj(:,:,60)=Obj(:,:,60).*3;
S= zeros(cnt_t,cnt_r);
for k = 1:cnt_t  
k
    for l = 1:cnt_r 
        S1=zeros(pnx,pny);
        for z_cnt = [30 60] 
            
            A = exp(1i*sqrt(  (tr_x(k)-pX).^2 + (tr_y(k)-pY).^2 + (pza(z_cnt)).^2 ) );
            B = exp(1i*sqrt(  (re_x(l)-pX).^2 + (re_y(l)-pY).^2 + (pza(z_cnt)).^2 ) );
            S1 = S1 + Obj(:,:,z_cnt).*A.*B; 
            
        end % for t
        
        S(k,l) = sum(sum(S1)); 
        
    end  % for l
end  % for k

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

请先登录，再进行评论。

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

Answer 1

Ishu 2024-4-4

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/814685-for-loops-vectorization-for-gpu#answer_1436301

在 MATLAB Online 中打开

Hi Youngjin,

I understand that you want to use GPU computation in MATLAB. To use GPU computation you can convert your normal arrays into "gpuArray".

% converting normal arrays into gpuArray
[pY, pX] = gpuArray(meshgrid(linspace(-0.5,0.5,pnx),linspace(-0.5,0.5,pny)));
cnt_t=2601;
cnt_r=2601;
tr_x=gpuArray(linspace(-1,1,cnt_t));
tr_y=gpuArray(linspace(-1,1,cnt_t));
re_x=gpuArray(linspace(-2,2,cnt_r));
re_y=gpuArray(linspace(-2,2,cnt_r));
pza=gpuArray(linspace(0,5,101));