search

ginkgo-project / ginkgo

Numerical linear algebra software package
https://ginkgo-project.github.io/
BSD 3-Clause "New" or "Revised" License
384 stars 86 forks source link

Add a batch CG solver and reference/omp kernels #1598

Closed pratikvn closed 1 month ago

pratikvn commented 2 months ago

This PR adds a batch::solver::Cg class, which similar to the batch::solver::Bicgstab enables solution for SPD matrices. The functionality is equivalent of the non-batched Cg solver.

Some comments

  1. The .cu/.hip.cpp/.dp.cpp files can be safely ignored. All paths leading to those should result in GKO_NOT_IMPLEMENTED. These device kernels will be added in another PR.
  2. The class and the framework is very similar to the bicgstab class, so hopefully it should be fairly easy to review.