OHDSI / bayes-bridge

Bayesian sparse regression with regularized shrinkage and conjugate gradient acceleration
https://bayes-bridge.readthedocs.io/en/latest/
18 stars 7 forks source link

Consider using GPUs for matrix operations #12

Closed chinandrew closed 2 years ago

chinandrew commented 2 years ago

Try something like cuSPARSE to help speed up matrix operations with GPUs.

aki-nishimura commented 2 years ago

I think it makes sense to close the issue for now, but just leaving a comment that the current GPU support is limited to a sparse design matrix for logistic model only.

chinandrew commented 2 years ago

Good point, we can open another issues to track the other tasks.