How to create a sequence of numbers on the GPU

3 次查看(过去 30 天)
I want to do the follow in a new GPU Array, efficiently:
myarray = 1:100000
gpuArray(1:100000) works of course, but if I'm not wrong it's creating the large array in main memory then moving the whole thing to the GPU. I would expect that I can create the sequence of numbers on the GPU directly more efficiently, as I can do with many random numbers.

采纳的回答

Edric Ellis
Edric Ellis 2016-1-21
The direct equivalent is to use gpuArray.colon, like so:
myarray = gpuArray.colon(1,100000);
More here in the doc about building gpuArrays.
  1 个评论
David Parks
David Parks 2016-1-21
编辑:David Parks 2016-1-21
Aha! Didn't see that one, thanks! Looks like it performs nearly the same as linspace, maybe 4ish percent faster.
>> tic; for i = 1:100000; gpuArray.linspace(1,100000,100000); end; toc;
Elapsed time is 9.975085 seconds.
>> tic; for i = 1:100000; gpuArray.colon(1,100000); end; toc;
Elapsed time is 9.486463 seconds.
Doesn't look like I can go back and edit the accepted answer after the fact, drat.

请先登录,再进行评论。

更多回答(1 个)

David Parks
David Parks 2016-1-21
Found it.
gpuArray.linspace(1,1000,1000);
>> tic; for i = 1:10000; gpuArray(1:100000); end; toc;
Elapsed time is 2.737965 seconds.
>> tic; for i = 1:10000; gpuArray.linspace(1,100000,100000); end; toc;
Elapsed time is 0.916697 seconds.

类别

Help CenterFile 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