Vectorize a series of interpn calculations with GPU

Question

deathtime 2023-5-19

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1968034-vectorize-a-series-of-interpn-calculations-with-gpu

回答： Meet 2024-9-9，4:50

Let's say I have a cell array with equal sized (but different) 4D datasets:

nPoints = 5;
dataCell = cell(1,nPoints);
 for i = 1:nPoints
     dataCell{i} = rand(3, 2, 4, 3);
 end

For the data arays in each cell, I would like to carry out a 4D interpn. Let's say the interpolation sample arrays and targets for each dimension are as follows:

dimOneArray = [0.1 0.2 0.3];
dimTwoArray = [2 4];
dimThreeArray = [1 2 3 4];
dimFourArray = [0.3 0.6 0.9];
dimOneTarget = 0.15;
dimTwoTarget = 3;
dimThreeTarget = 2.5;
dimFourTarget = 0.4;

Then I would like to carry out nPoints interpn calculations in a vectorized manner i.e. I would like to produce the same result as below (interpArray) without a for loop:

interpArray = nan(nPoints, 1);
for i = 1:nPoints
    interpArray(i) = interpn(dimOneArray, dimTwoArray, dimThreeArray, dimFourArray, cell2mat(dataCell(i)), dimOneTarget, dimTwoTarget, dimThreeTarget, dimFourTarget);
end

How would I use the GPU to carry out this calculation?

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

请先登录，再进行评论。

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

Answer 1

Meet 2024-9-9，4:50

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1968034-vectorize-a-series-of-interpn-calculations-with-gpu#answer_1512939

在 MATLAB Online 中打开

Hi,

To perform interpolation of the given data on a GPU, you can perform the following steps:

Transfer your “dataCell” array to the GPU using the “gpuArray” function.
Initialize an empty array on the GPU to store the final interpolation results.
Define a function handle that will be used to perform the interpolation.
Use “arrayfun” function to apply the interpolation to each element of the “dataGPU” array.

% Convert dataCell array to gpuArray
dataGPU = cellfun(@gpuArray, dataCell, 'UniformOutput', false);
% Initialize result array which stores the result of the computation
interpArrayGPU = gpuArray.nan(nPoints, 1);
% Defining a function handle for interpolation
interpolateFunction = @(data) interpn(dimOneArray, dimTwoArray, dimThreeArray, dimFourArray, data, dimOneTarget, dimTwoTarget, dimThreeTarget, dimFourTarget);
% Using arrayfun to apply the interpolation function to each element of
% dataGPU array
interpArrayGPU = arrayfun(@(i) interpolateFunction(dataGPU{i}), 1:nPoints);