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에 대해 자세히 알아보기
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!