Doesn't the GPU Coder support singleton expansion of ARRAYFUN?
1 次查看(过去 30 天)
显示 更早的评论
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)
0 个评论
采纳的回答
Joss Knight
2021-10-2
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 个)
另请参阅
类别
在 Help Center 和 File Exchange 中查找有关 Kernel Creation from MATLAB Code 的更多信息
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!