cupy / cupy/cupy

Support for batched `cupyx.scipy.sparse.linalg.spsolve`

Open
#6,509 8 comments 0 reactions 0 assignees View on GitHub
issue-checked
Dominant language
Python
Stars
12.3k
Forks
1.1k
Avg merge
1d 20h
Merged PRs (30d)
45

Description

### Description

Currently, `cupy.linalg.solve` supports batch processing. Such functionality for `cupyx.scipy.sparse.linalg.spsolve` would be a great asset and would prevent slow, manual loops.

For example:
`N` being the batch dimension
`A x = b`
, with
- A: Sparse matrix with dimension `(N, M, M)` or as list with `N`-Sparse matrices `(M, M)`.

- b: Dense matrix with dimension `(N, M)` or as list with `N`-Dense vectors `(M)`.
- x: Dense matrix with dimension `(N, M)` or as list with `N`-Dense vectors `(M)`.

### Additional Information

_No response_

Contributor guide

Open the contributing guide

Research direction

Start at the `cupyx.scipy.sparse.linalg.spsolve` entry point named in the issue, then compare its behavior with batched `cupy.linalg.solve`. Done means supporting the listed batched or list-of-matrices forms for A and b and returning corresponding dense solutions.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
hpc
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.