Adding just one line in my code slows the GPU.

1 次查看（过去 30 天）

TAEYOON KIM 2020-2-18

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/506220-adding-just-one-line-in-my-code-slows-the-gpu

评论： TAEYOON KIM 2020-2-19

hello, my first code was

for i = 1:1000
    r=0.8*r+0.2*tanh(A*r+W_in*xc(i+T0)+(0.3)*bias);  
end

In the code above,the gpu is faster than cpu but I have to stack 'r' which is (nx1) matrix, so I just adding 'R' on previous code makes the gpu slower than cpu.

R=zeros(n,1000);
for i = 1:1000
    r=0.8*r+0.2*tanh(A*r+W_in*xc(i+T0)+(0.3)*bias);
    R(:,i)=r;
end

I want to make my code runs fast on gpu than cpu. What should I do??

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

请先登录，再进行评论。

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

回答（1 个）

Matt J 2020-2-18

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/506220-adding-just-one-line-in-my-code-slows-the-gpu#answer_416184

编辑：Matt J 2020-2-18

在 MATLAB Online 中打开

Pre-allocate on the GPU. Also, pre-compute things on the GPU that are easily vectorized and don't depend on r.

R=gpuArray.zeros(n,1000);
r=gpuArray(r);
A-gpuArray(A);
increments = W_in*gpuArray( xc((1:1000)+T0)  ) + 0.3*bias;
for i = 1:1000
    r=0.8*r+0.2*tanh(A*r+ increments(i));
    R(:,i)=r;
end