How to set all one gpuArray in mex?

wei zhang

2021 2 16

1 个回答

回答已采纳

8 次查看（30 天）

显示更早的评论

0 个投票

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更早的评论

请先登录，再进行评论。

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

请先登录再关注

采纳的回答

Edric Ellis 2021-2-16

0 个投票

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').