parallel-processing - 有什么方法可以在matlab中为spmd命令添加开关？

Question

我正在使用 matlab2011 进行多核并行计算，这在我的代码中只是使用块 SPMD END 实现的。但是，有一段时间，我想根据输入参数关闭程序中的spmd。我尝试以下代码，但它不起作用

if (switchSpmdOn)
  spmd
end
% here I put my code for calculation in parallel or series
if (switchSpmdOn)
  end % this end is used to close the spmd block
end

我想知道代码中是否有类似 marco 的东西来关闭 spmd。

Answer 1

您可以将工作人员的数量作为参数传递给spmd. 如果指定的工人数量是0，即，

spmd(0)
   statement; %# block body
end

然后 MATLAB 将在本地执行块体并创建复合对象，就像没有可用的池一样。

例子

switchSpmdOn = true;

%# set the number of workers
if (switchSpmdOn)
    workers = 3; 
    matlabpool(workers);
else
    workers = 0;
end

%# application. if 'workers' is 0, MATLAB will execute this code locally
spmd (workers)
    %# build magic squares in parallel
    q = magic(labindex + 2);
end

for ii=1:length(q)
  % plot each magic square
  figure, imagesc(q{ii});
end

if (switchSpmdOn)
    matlabpool close
end