PARFOR: Broadcast variable

18 次查看(过去 30 天)
Hi, I'm starting to use "parfor" in my MATLAB scripts.
In this simple example, for each iteration, I get a 2D 'sub_matrix' from a 2D big matrix (image).
But I got this warning message: "the entire array is a broadcast variable. This might result in unnecessary communication overhead.
% For each pixel in the 'image':
parfor row = 1:N
for col = 1:M
% Crop a sub matrix from the original image:
sub_image = image(row:row+H-1, col:col+W-1); % WARNING HERE!
% Do other stuffs...
end
end
How can I modify this code to avoid this communication overhead?
  5 个评论
Walter Roberson
Walter Roberson 2020-1-14
The Parallel Computing Toolbox is the same toolbox that handles parfor and handles GPU, but they work in very different ways.
GPU performance is much reduced by indexing, and only really wins out when you have operations that can be vectorized over an entire array. For your code, that would mean continually creating new gpuArray for each subimage, and then what the GPU would accelerate would be the processing over the subimage. But that would still involve a lot of memory transfer.
You might be thinking that you could send the entire large image to GPU and create the subimages on there, but the computation engines work a bit oddly.
Computation on an NVIDIA machine is divided up into compute controllers. Each compute controller can be executing a different set of instructions than other compute controller are executing. Each compute controller is responsible for a number of computation cores. The compute controller decodes an instruction, and sends the same instruction to each of the computation cores under its control. Each computation core then executes that same instruction for its data. The way conditional execution works is not by having different compute instructions executed by different compute cores: instead, a mask is created, one per compute core, and any compute core for which the mask is not true, idles instead of executing the instruction.
So for example,
subimage = image(1:15, 1:20)
would involve implicitly creating a mask the size of image that was true for positions in the 15 x 20 upper corner, and a transfer instruction would be executed, and the compute nodes with mask true would execute the transfer instruction and the other compute nodes would idle themselves for that instruction -- compute nodes for the entire array are involved, with most of them idling.
This is very different than CPU programming, where on CPU programming it is almost always more efficient to restrict your computation to only the locations that need to be processed; on GPU you would rather have entire arrays being processed unconditionally so that you do not waste compute cores.
Walter Roberson
Walter Roberson 2020-1-14
I would point out, by the way, that if you were to replicate your template several times in each direction, that you could process a corresponding sized chunk of the image. You still need to shift the window around, but you would be doing more in each chunk -- better vectorization.

请先登录,再进行评论。

采纳的回答

Walter Roberson
Walter Roberson 2020-1-14
parfor row = 1:N
for col = 1:M
% Crop a sub matrix from the original image:
sub_image = image(row:row+H-1, col:col+W-1); % WARNING HERE!
In order for it to be possible to slice the variable, one of the dimensions of the variable would have to depend only on the parfor index (possibly plus a constant.) You can do slicing by forming separate variables:
image_slices = cell(N,1);
for row = 1 : N
image_slices{row} = image(row:row+H-1, 1:M+W-1);
end
parfor row = 1 : N
image_slice = image_slices{row};
for col = 1 : M
sub_image = image_slice(:, col:col+W-1);
...
end
end
This would of course end up using a heck of a lot of memory.
I would suggest that you might be better off rewriting everything in terms of a 2D filter operation or possibly nonlinear filtering.
  4 个评论
Walter Roberson
Walter Roberson 2020-1-15
You are using all of template in every worker. MATLAB must send all of it over: there is no getting around making it a broadcast variable.
Note: for increased efficiency you can pre-calculate template(logical_idx) and store it in a variable instead of calculating it in your inner loop.
Walter Roberson
Walter Roberson 2020-1-15
image_slices{row} = image(row:row+template_height-1, 1:template_width-1);
The 1:template_width-1 is wrong. The upper bound needs to be the such that col:col+template_width-1 fits for the largest col value -- so the upper bound must be search_width+template_width-1

请先登录,再进行评论。

更多回答(1 个)

Matt J
Matt J 2020-1-14
编辑:Matt J 2020-1-14
The code looks suspiciously like an attempt at weighted normalized cross-correlation. If so, see here for a possibly faster alternative to parfor,
  3 个评论
Matt J
Matt J 2020-1-14
编辑:Matt J 2020-1-14
The function uses a combination of fft2 and conv2, which are parallelized internally by Matlab, to do most of the computations.
Nycholas Maia
Nycholas Maia 2020-1-14
Great! Thank you again!

请先登录,再进行评论。

类别

Help CenterFile Exchange 中查找有关 Matrix Indexing 的更多信息

Community Treasure Hunt

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

Start Hunting!

Translated by