如何使用 GPU/MATLAB 加速自定义 function？

Question

假设我有一个 function f(x) ,st yi=f(xi)

现在给定一个array=(x1,x2,……,xn)，我们要计算z = f(array)，如何使用GPU加速计算？

即代码没有体现gpu的并行加速能力

shape = 40;
shapez = 100000;
b = gpuArray(single(ones(shape,shape,shapez)));
for i=1:shapez
    c =mean(b(:,:,i));
end
disp(['gpu processing time: ', num2str(toc(t1)), ' seconds']);

如果我按如下方式更改代码，它会变得更快。

t1 = tic;
shape = 40;
shapez = 100000;
b = gpuArray(single(ones(shape,shape,shapez)));

disp(['gpu processing time: ', num2str(toc(t1)), ' seconds']);

那么，如果我的算法的基本逻辑如下，如何实现它才能被GPU加速呢？

for i=1:length(array)
    yi = f(xi)
end

Answer 1

要使用预先存在的 Matlab GPU 加速功能，您可以在此处找到它们

如果您想实现自己的功能，可以按照官方文档中的描述使用 MEX。