Accelerate eigs with GPU
显示 更早的评论
Dear all,
I have implemented a numerical solver (of the Fokker-Planck equation) in MATLAB.
At some point, the algorithm needs to calculate
eigs(L, 1, 0)
of a very large sparse matrix L.
I would like to perform this calculation on a GPU to lower the computational costs. So, I created the array
Lgpu = gpuArray(L);
on the GPU and tried to calculate
eigs(Lgpu, 1, 0)
again.
Unfortunately, I receive the error message: "First argument must be a double matrix or a function."
I am wondering what the cause of this error might be and appreciate any help from you.
Thank you very much.
best,
Sven
回答（2 个）
Christine Tobler
2020-7-30
编辑：Edric Ellis
2020-7-30
2 个投票
The eigs function is not supported on the GPU. There is support for sparse matrices on the GPU, since R2015a: Release notes parallel computing toolbox.
You could pass a function handle to EIGS that would use computation on the GPU, but would need to accept and pass back out vectors on the CPU. I'm not sure how efficient that would be, but it could be worth a try.
6 个评论
Sven Auschra
2020-7-30
Bruno Luong
2020-7-30
编辑：Bruno Luong
2020-7-30
I posted it yesterday but I remove it due to Walter post.
But it's no complicated than this.
eigs( @(x) gather(Lgpu\gpuArray(x)), size(L,1), 1, 0);
or
eigs(@(x) gather(bicg(Lgpu,gpuArry(x))), size(L,1), 1, 0);
Christine Tobler
2020-7-30
To compute the largest eigenvalue by absolute value, you would use
n = size(L, 1);
eigs(@(x) gather(Lgpu*x), n, 1)
which would apply the matrix-vector product on the GPU, and then move back to the CPU to pass that result back to EIGS. This could give you a first impression if there's anything to gain from using this as opposed to just calling
eigs(L, 1);
Since you're actually computing the eigenvalues of L that are closest to zero, things are a bit more complicated, and I'm still investigating if this can be done efficiently on the GPU at the moment.
Christine Tobler
2020-7-30
Bruno, it looks like we were posting at the same time there. Using Lgpu\gpuArray(x) will definitely work, but since backslash is called many times here, to be efficient Lgpu should be factorized and this factorization used in the function handle. Otherwise I'd expect that the CPU version will still be faster.
Unfortunately, LU doesn't support sparse gpuArrays at this point, which means we have to call backslash which will compute this factorization underneath on every call.
Overall, it's worth trying out these ideas, but I kind of expect that while they will move some computation to the GPU, they won't actually make your EIGS call faster than the CPU version, because gpuArray doesn't yet support all tools to get the best possible performance out of EIGS.
Sven Auschra
2020-7-30
Bruno Luong
2020-7-30
Yeah I would also expect any speedup using GPU. Mostly EIGS on sparse is mainly an iterative process in double layers. Nothing really leans for GPU computation.
Walter Roberson
2020-7-29
1 个投票
There is no GPU support for sparse arrays.
类别
在 帮助中心 和 File Exchange 中查找有关 GPU Computing 的更多信息
2020-7-29
2020-7-30
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!
Translated by 
