How to parallelize many small SVD's on a GPU

13 次查看(过去 30 天)
I'm computing SVD's of millions of small (e.g. 5 x 10) matrices. It is surprisingly fast as a for-loop on the CPU, however I need the computation to be even faster and would like to know if parallelization can be leveraged on a GPU. SVD doesn't currently appear to be a function supported by pagefun for gpuArray objects. I'm using R2020a and the Parallel Computing toolbox. Thanks

采纳的回答

Christine Tobler
Christine Tobler 2021-12-9
As of R2021b, there is a function pagesvd which is supported on both CPU and distributed arrays. It's not supported for gpuArray as of now.
  7 个评论
Joss Knight
Joss Knight 2021-12-18
gpuArray/pagesvd coming next year, watch this space...

请先登录,再进行评论。

更多回答(1 个)

Matt J
Matt J 2021-12-9
编辑:Matt J 2021-12-9
SVD doesn't currently appear to be a function supported by pagefun for gpuArray objects
Perhaps not for gpuArrays, but it is supported for distributed arrays:

类别

Help CenterFile Exchange 中查找有关 Linear Algebra 的更多信息

产品


版本

R2020a

Community Treasure Hunt

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

Start Hunting!

Translated by