How to set all one gpuArray in mex?

Question

wei zhang 2021-2-16

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/746462-how-to-set-all-one-gpuarray-in-mex

回答： Edric Ellis 2021-2-16

Hi,

I need to set a all-one double gpuArray. I think I could use mxGPUCreateGPUArray to initial it. But it seems mxGPUCreateGPUArray could only initial a all-zero array like below.

mxGPUArray *x = mxGPUCreateGPUArray(ndim, dims, mxDOUBLE_CLASS, mxREAL, MX_GPU_INITIALIZE_VALUES);

Is there any good way to do this? Do I need to write a kernel function to do it with cuda function? Thank you.

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

请先登录，再进行评论。

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

Answer 1

Edric Ellis 2021-2-16

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/746462-how-to-set-all-one-gpuarray-in-mex#answer_624472

You basically have two options: either create and uninitialized mxGPUArray and fill the values using CUDA code (either a kernel, or one of the cuMemset* variants if your data type is small enough), or pass in to your MEX function an array of ones that you create in MATLAB using ones(...,'gpuArray').