Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

Batch Solving in cupy.linalg.eigh

See original GitHub issue

The interface to numpy’s hermitian eigenvalue solver numpy.linalg.eigh may accept an (n, m, m) array for batch calculations. CuSOLVER has the corresponding functions cusolverDnSsyevjBatched, cusolverDnDsyevjBatched, cusolverDnCheevjBatched, and cusolverDnZheevjBatched. It seems like it would be a good idea to generalize cupy’s interface to match what numpy has. I have never contributed to cupy, but have made similar changes to a julia GPU library. Is this a good feature to add and would this be a good first time PR for me to make? I have a need for a function like this and so I have some motivation to get it done.

Issue Analytics

State:
Created 3 years ago
Reactions:1
Comments:8 (5 by maintainers)

Top GitHub Comments

2reactions

electronsandstuffcommented, Jul 1, 2020

Sweet! If you have some code already written, then it’s all yours.

1reaction

leofangcommented, Jul 28, 2021

This can be closed.

Top Results From Across the Web

cupy.linalg.eigh — CuPy 11.4.0 documentation

Return the eigenvalues and eigenvectors of a complex Hermitian (conjugate symmetric) or a real symmetric matrix. Returns two objects, a 1-D array containing...

cupy.linalg.eigvalsh — CuPy 11.4.0 documentation

Compute the eigenvalues of a complex Hermitian or real symmetric matrix. Main difference from eigh: the eigenvectors are not computed. ... This function...

cupy.linalg.eigvalsh — CuPy 8.6.0 documentation

Calculates eigenvalues of a symmetric matrix. This method calculates eigenvalues a given symmetric matrix. Note that cupy.linalg.eigh() calculates both ...

Linear algebra (cupy.linalg) — CuPy 11.4.0 documentation

Returns the sum along the diagonals of an array. Solving equations and inverting matrices#. linalg.solve (a, ...

cupy.linalg.eigh — CuPy 4.5.0 documentation

cupy.linalg.eigh¶ · a (cupy.ndarray) – A symmetric 2-D square matrix. · UPLO (str) – Select from 'L' or 'U' . It specifies which...