Skip to content

Sparse linear solve: CG/GMRES on trnsparse matrices #23

Description

@scttfrdmn

When the sparse matvec path in trnsparse matures, cg and gmres should accept sparse A directly. Currently both treat A as dense (torch.Tensor) or as a callable matvec.

Scope

  • Extend the A argument type to Union[torch.Tensor, Callable, trnsparse.SparseMatrix] (or whatever the trnsparse sparse type is called at the time)
  • Dispatch to trnsparse.spmv for the matvec path
  • Add sparse-A tests

Depends on

  • trnsparse providing a stable public matvec API (cross-project)

Metadata

Metadata

Assignees

No one assigned

    Labels

    iterativeCG / GMRES iterative solversphase-3-perfRoadmap: phase-3-perf

    Type

    No type

    Projects

    No projects

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions