并行化 Matlab 中的嵌套循环

Question

我正在尝试加快Matlab中某些面板数据的模拟。 我必须首先模拟个人（循环索引 ii 从 1 到 N），然后模拟每个超过年龄的人（循环索引 jj 从 1 到 JJ）。 代码很慢，因为在两个循环内有一个双线性插值要做。

由于外循环中的迭代是独立的，我尝试在外循环（由ii索引的循环）中使用parfor ，但我得到错误消息“由于变量hsim的使用方式，parfor无法运行”。 如果可能，有人可以解释为什么以及如何解决问题吗？ 任何帮助是极大的赞赏！

a_sim = zeros(Nsim,JJ);
h_sim = zeros(Nsim,JJ);
% Find point on a_grid corresponding to zero assets
aa0 = find_loc(a_grid,0.0);
% Zero housing
hh0 = 1;
a_sim(:,1) = a_grid(aa0);
h_sim(:,1) = h_grid(hh0);
parfor ii=1:Nsim !illegal
    for jj=1:JJ-1
        z_c = z_sim_ind(ii,jj);
        apol_interp = griddedInterpolant({a_grid,h_grid},apol(:,:,z_c,jj));
        hpol_interp = griddedInterpolant({a_grid,h_grid},hpol(:,:,z_c,jj));
        a_sim(ii,jj+1) = apol_interp(a_sim(ii,jj),h_sim(ii,jj));
        h_sim(ii,jj+1) = hpol_interp(a_sim(ii,jj),h_sim(ii,jj));
    end
end

Answer 1

我认为@Ben Voigt 的建议是正确的。 拼写出来，做这样的事情：

parfor ii=1:Nsim
    a_sim_row = a_sim(ii,:);
    h_sim_row = h_sim(ii,:);
    for jj=1:JJ-1
        z_c = z_sim_ind(ii,jj);
        apol_interp = griddedInterpolant({a_grid,h_grid},apol(:,:,z_c,jj));
        hpol_interp = griddedInterpolant({a_grid,h_grid},hpol(:,:,z_c,jj));
        a_sim_row(jj+1) = apol_interp(a_sim_row(jj),h_sim_row(jj));
        h_sim_row(jj+1) = hpol_interp(a_sim_row(jj),h_sim_row(jj));
    end
    a_sim(ii,:) = a_sim_row;
    h_sim(ii,:) = h_sim_row;
end

这是一个相当标准的parfor模式来解决这个限制（在这种情况下， parfor无法发现你正在做的事情就外循环而言不是顺序无关的） - 提取整个切片，做任何需要的事情, 然后把整个切片放回去。