Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add GPU Cholesky Decomposition #1058

Closed
rok-cesnovar opened this issue Oct 29, 2018 · 0 comments
Closed

Add GPU Cholesky Decomposition #1058

rok-cesnovar opened this issue Oct 29, 2018 · 0 comments
Assignees
Labels
Milestone

Comments

@rok-cesnovar
Copy link
Member

Description

Adds the GPU Cholesky decomposition for the matrix_gpu type, and adds the GPU functionalites to the /prim and /rev implementations of the Cholesky Decomposition. This issue will be tackled with 2 pull requests:

  • the kernel code for small blocks, the cholesky_decompose(matrix_gpu&) function and /prim implementation
  • /rev implementation

Example

Use of cholesky_decompose(matrix_gpu&):

stan::math::matrix_gpu m0(m0_eigen);
using stan::math::cholesky_decompose;
stan::math::matrix_gpu foo = cholesky_decompose(m0);

Use of cholesky_decompose(stan::math::matrix_d) and cholesky_decompose(stan::math::matrix_v):

using stan::math::cholesky_decompose;
stan::math::matrix_gpu foo = cholesky_decompose(m0_eigen);

Expected Output

The expected output is faster execution of Cholesky Decomposition.

Additional Information

/

Current Math Version

v2.18.0

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

No branches or pull requests

3 participants