通过Matlab GPU调用arrayfun时如何记录循环的迭代变量?

时间:2018-12-16 05:00:35

标签: matlab gpu-programming

在下面的附加代码中,我想在while循环中将变量 Z 记录为数组,以进行进一步处理,例如,在数组中查找Z的最大值。但是Matlab 2017b不支持数组索引,即使使用了嵌套函数也是如此。有人可以向我提供详细信息来解决这个问题吗?

clear all
clc

maxIterations=100;
gridSize=300;
xlim=[-0.75, -0.73];
ylim=[ 0.12,  0.14];
t=tic();
x=gpuArray.linspace(xlim(1), xlim(2), gridSize);
y=gpuArray.linspace(ylim(1), ylim(2), gridSize);
[xGrid,yGrid]=meshgrid(x,y);
Pos=gpuArray.zeros(maxIterations,1);
count=parent_fun(xGrid,yGrid,maxIterations,Pos);
count=gather(count);
gpuArrayfunTime=toc(t)
figure(1)
imagesc(x,y,count)
reset(gpuDevice(1))

嵌套函数的定义如下。

function result=parent_fun(xGrid,yGrid,maxIterations,Pos)
         function count=tar_fun(x0,y0)
                  z0=complex(x0,y0);
                  z=z0;
                  count=1;
                  while (count<=maxIterations) && (abs(z)<=2)
                         count=count+1;
                         z=z*z+z0;
                         Pos(count,1)=z;  % where error arises!
                  end 
                  count=max(log(count),log(abs(z)));
         end
result=arrayfun(@tar_fun, xGrid, yGrid);
end

0 个答案:

没有答案