Doesn't the GPU Coder support singleton expansion of ARRAYFUN?
显示 更早的评论
I have a question about SINGLETON EXPANSION for ARRAYFUN on GPU.
For gpuArrays, @times with ARRAYFUN and direct calculation on M-file both work fine.
However, MEX generated by GPU Coder from a code using ARRAYFUN does not seem to perform SINGLETON EXPANSION well.
Doesn't the GPU Coder support SINGLETON EXPANSION of ARRAYFUN?
Shogo MURAMATSU
Sample code
function Y = arrayfuntest(m,X) %#codegen
Y = arrayfun(@times,m,X);
%Y = m.*X;
end
Code generation for ARRAYFUN
Code generation for a direct operation
Preparation of arrays
m = gpuArray(single([1;-1;1]))
X = gpuArray(rand(3,'single'))
Execution with M-file
Y = arrayfuntest(m,X)
MEX Generation
cfg = coder.gpuConfig('mex');
nChs = 64;
datatype = 'single';
aM = coder.typeof(gpuArray(cast(0,datatype)),[nChs 1],[1 0]);
aX = coder.typeof(gpuArray(cast(0,datatype)),[nChs nChs],[1 1]);
cfg.DynamicMemoryAllocation = 'Off';
cfg.GenerateReport = true;
codegen -config cfg -o arrayfuntest_mex arrayfuntest.m -args { aM, aX }
Execution with MEX-file
Y = arrayfuntest_mex(m,X)
采纳的回答
Joss Knight
2021-10-2
0 个投票
Support for dimension expansion is actually restricted to the GPU version of arrayfun, and GPU Coder supports the CPU version's syntax. So the answer is no, it doesn't support dimension expansion.
更多回答（0 个）
类别
在 帮助中心 和 File Exchange 中查找有关 GPU Computing in MATLAB 的更多信息
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!
Translated by 
