Vectorised a nested conditional function for gpuarray & arrayfun

Question

Sendy Phang 2014-10-9

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/158069-vectorised-a-nested-conditional-function-for-gpuarray-arrayfun

回答： Daniel Simmons 2014-10-10

Hi, I have a class which has many arrays and a nested functions.

The goal is I want to make all the arrays as gpuarray and use arrayfun, to be run in GPU, thus I'd like to have them well vectorised. Any thought? Thanks

%%the class
classdef egclass < handle
      properties
          cond = [1 1 1 0 0]; 
          x = [10 20 30 40 50];
          y = [0 0 0 0 0];
          out= [0 0 0 0 0]; 
      end
  end
%%the function 
function doOperation(eg)
      [nR,nC] = size(eg.cond);
      %%need to be vectorised!!!
      for i = 1:nC
          if eg.cond(i) == 0
              doTimes(eg,i);
          end
          if eg.cond(i) == 1  
              doAdd(eg,i);
          end
      end
  end
function doTimes(eg,i)
    eg.out(i) = eg.x(i) * eg.y(i);
end
function doAdd(eg,i)
    eg.out(i) = eg.x(i) + eg.y(i);
end
%%debug.m
clear all; clc;
eg = egclass;
doOperation(eg);

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

Daniel Simmons 2014-10-10

1
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/158069-vectorised-a-nested-conditional-function-for-gpuarray-arrayfun#answer_154681

在 MATLAB Online 中打开

Hi Sendy,

If you're trying to use arrayfun to pass the arrays to the GPU inside the class, this is no problem! Just use this as your member function:

function doOperation(eg)
  gpu_x = gpuArray( eg.x );
  gpu_y = gpuArray( eg.y );
  gpu_cond = gpuArray( eg.cond );
  eg.out = arrayfun( @doAddOrTimes, gpu_cond , gpu_x, gpu_x );
end

where the "doAddOrTimes" function will incorporate the "if" statement. Of course, this function, and any others used on the GPU, will have to be an external function with no class attributes. As far as I'm aware, manipulating classes on the GPU is not possible.

Dan