help vectorize code for GPU

3 次查看（过去 30 天）
Octavian
Octavian 2015-4-5
编辑： Matt J 2015-4-5
Dear All,
I have a for loop code snippet which in essence, involves 2D matrix or elementwise multiplication, and 2D matrix addition, for 2D sheets along the 3rd dimension in 3D double arrays. Here is what I mean:
for i=1:M
A(:,:,i)=B(:,:,i)* C(:,:,i);
D(:,:,i)=E(:,:,i).*F(:,:,i);
G(:,:,i)=H(:,:,i)+I(:,:,i);
end
Besides using parfor, is there a way to vectorize this code for GPU use to speed it (A-I are huge matrices).
Thank you, as always, Octavian

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

采纳的回答

Matt J
Matt J 2015-4-5
编辑：Matt J 2015-4-5
I assume all the arrays shown are gpuArrays. You can use pagefun for creating A
A=pagefun(@mtimes, B,C);
There is also mtimesx ( Download ) on the File Exchange which optimizes this kind of thing for CPU variables.
The remaining operations are already internally vectorized, both for ordinary CPU array variables and gpuArrays. You can just do
D=E.*F;
G=H+I;

更多回答（0 个）

类别

在 Help Center 和 File Exchange 中查找有关 GPU Computing 的更多信息

标签

Community Treasure Hunt

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

Start Hunting!

Translated by