GPU memory usage for Hadamard product

79 次查看(过去 30 天)
Massimiliano Zanoli
Massimiliano Zanoli 2024-7-12,9:10
评论: Joss Knight 2024-7-18,14:41
I have a GPU with 48 Gb of RAM.
I have a large matrix A (complex single 45927x45927, gpuArray) taking 16 Gb of my GPU.
I get to the following Hadamard product with a vector B (complex single 45927x1, gpuArray) with approximately 28 Gb of free RAM on the GPU:
C = A .* (B.')
But this throws an "out of memory on the device" error, despite there being enough memory for C (also 16 Gb). Why is this happening? Does MATLAB implicitly convert B to a full matrix? Is there a way to get this multiplication to work on the GPU within the available memory? (maybe this has been fixed in later releases of MATLAB?)
  8 个评论
Matt J
Matt J 2024-7-18,13:33
编辑:Matt J 2024-7-18,13:34
What happens if you do other operations, like,
C=A+A
C=A+B
C=gpuArray.zeros(size(A))+1j*gpuArray.ones(size(A));

请先登录,再进行评论。

回答(4 个)

Sivsankar
Sivsankar 2024-7-12,10:44
Hi,
I believe that you are probably limited by the size of RAM of your GPU. I can suggest some ideas that you can implement which may work.
  • Upgrade to the latest version of MATLAB.
  • Pre-allocate the memory for the resultant gpuArray matrix C.
  • Clear any existing gpuArray that may take up the RAM.
  • Reset GPU device to release memory.
You can leverage this MATLAB answer that highlights a similar issue.
You can also make use of this MathWorks documentation that explains on how to resolve such “out of memory” errors.
Hope this works for you. Thanks.
  1 个评论
Massimiliano Zanoli
Massimiliano Zanoli 2024-7-14,12:24
Thanks for the suggestions. I tried clearing and pre-allocating, but MATLAB still seemingly attempts to allocate 3 * 16 Gb for the operation, even though it should take only 2 * 16 Gb. Is it meant to be that way? Otherwise, I will try with a newer version of MATLAB to see if this has been fixed.

请先登录,再进行评论。


Matt J
Matt J 2024-7-14,14:28
This is just a guess, but possibly the GPU memory is fragmented. I.e., maybe your available 28GB is broken into blocks that are smaller than 16GB and therefore Matlab cannot find enough space to allocate C.

Catalytic
Catalytic 2024-7-14,19:32
编辑:Catalytic 2024-7-14,19:40
Does MATLAB implicitly convert B to a full matrix?
When you say this, are you implying that issparse(B)=true? If so, then it is possible that C could consume up to 32 GB.
  2 个评论
Matt J
Matt J 2024-7-14,19:52
I don't think that would be possible. If B were sparse, you would get a different error if you tried to multiply it with a single matrix.
Massimiliano Zanoli
Massimiliano Zanoli 2024-7-15,9:34
No, I improperly meant "full" as in matrix instead of vector, sorry for the confusion. I was wondering whether B is repeated from 45927x1 to 45927x45927, that could explain the memory requirement.

请先登录,再进行评论。


Joss Knight
Joss Knight 2024-7-18,11:52
编辑:Joss Knight 2024-7-18,11:54
I can't reproduce this.
You say you have 48Gb of GPU memory available...have you checked this? Try running gpuDevice and looking at the output. Maybe you have other variables taking up space. Maybe you have another MATLAB running and using the GPU?
Also, did it work the first time you ran it but not the second? If C already exists then you would need space for A, C and the new C all at once, and you would run out of memory.
  3 个评论
Joss Knight
Joss Knight 2024-7-18,14:08
Your gpuDevice output seems to indicate you have quite an old version of MATLAB. What version are you on?
Joss Knight
Joss Knight 2024-7-18,14:41
Never mind. I can't reproduce this in any old version, on Linux or Windows.
Just to be sure, make sure your script is standalone and is the only thing you run after you've started MATLAB. It should contain no other computation than creating A and B and then computing C, so for instance:
n = 45927;
proto = gpuArray(single(1i));
A = randn(n,'like',proto);
B = randn(n,1,'like',proto);
C = A.*(B.');
gpuDevice
whos

请先登录,再进行评论。

类别

Help CenterFile Exchange 中查找有关 GPU Computing 的更多信息

产品


版本

R2021a

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by