[FEA] Lanczos solver v2#2481
Conversation
|
/ok to test |
|
/ok to test |
|
/ok to test |
|
/ok to test |
|
/ok to test |
| void lanczos_solver( \ | ||
| const raft::resources& handle, \ | ||
| raft::sparse::solver::lanczos_solver_config<ValueType> config, \ | ||
| raft::device_vector_view<IndexType, uint32_t, raft::row_major> rows, \ |
There was a problem hiding this comment.
Here too- please consolidate this list by using device_coo_matrix_view.
cjnolet
left a comment
There was a problem hiding this comment.
We need to get this PR in because code freeze is looming so I'm merging this now. @lowener can we please follow-up by exposing the device_coo_matrix_view in the public API insted of the individual arrays? It should be a 1-2 line change.
|
/merge |
|
@cjnolet I think that the name of the parameters ( |
This reverts commit 64c09d7.
I unfortunately don't have permissions to push on @aamijar branch for the previous Lanczos solver PR (#2416) so I kept his commits and continued it here.
Lanczos Solver for Sparse Eigen Decomposition
We propose a new lanczos solver in raft that fixes the issues present in the previous solver
raft::sparse::solver::detail::computeSmallestEigenvectors.Specifically we address the following issues:
This new implementation is taken from the cupy library
cupyx.scipy.sparse.linalg.eigshwhere the thick-restart and full reorthogonalzation methods are used.Additionally this PR exposes a python api for raft lanczos solver with an interface similar to
scipy.sparse.linalg.eigshandcupyx.scipy.sparse.linalg.eigsh.