在 Matlab 中通过多个 GPU 处理大型密集矩阵

Question

只是矩阵 A 乘以矩阵 B（以及其他简单的运算），但每个矩阵（A 或 B）对于单个 GPU 来说都很大，无法存储。

A = rand(10000,10000)
A = distributed(A)
spmd
%     gpuDevice
    A = gpuArray(A)
    
end

我的解决方案失败了，因为“GPU 阵列仅支持基本的数字或逻辑数据类型”。

如果我手动将矩阵的一部分分配给每个GPU，我必须自己处理GPU（矩阵的每个部分）之间的通信时间，而且，我不能直接使用很多函数，例如eig或max。

我不知道matlab是否已经有一些解决方案（就像使用分布式数组来处理每个工作之间的通信时间一样）来处理这个问题。