Implementing batched mldivide on the GPU

2 次查看（过去 30 天）
Rodrigo
Rodrigo 2014-3-19
回答： Edric Ellis 2014-12-10
Hi, I need to solve A*x=B for size(A)~[10,10,1e5] and size(B)~[10,1,1e5], and it seems like something like pagefun would be the way to do it: except that mldivide is not an option. I understand that cuBLAS has had the ability to run batch LU decompositions for some time and the newer cublas also seems to have a batched solver.
Is there a reasonably simple way to implement this in MATLAB (preferably R2013b or earlier)?
Thanks

请先登录，再回答此问题。

采纳的回答

Edric Ellis
Edric Ellis 2014-3-19
Unfortunately, as you have found PAGEFUN does not yet support batched MLDIVIDE. You might be able to use the GPU Mex interface to call CUBLAS yourself.
  2 个评论
Rodrigo
Rodrigo 2014-3-19
I was hoping I wouldn't have to do this, but I suppose there's no better alternative yet.
Adam
Adam 2014-4-29
Just chiming in: I also have a need for a pagefun 'mldivide'. If someone has done the MEX implementation already, please link. It would be greatly appreciated.

请先登录，再进行评论。

更多回答（1 个）

Edric Ellis
Edric Ellis 2014-12-10
This is now supported in R2014b - see the release notes.

类别

在 Help Center 和 File Exchange 中查找有关 GPU Computing 的更多信息

标签

产品

Community Treasure Hunt

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

Start Hunting!

Translated by