Skip to content

Conversation

copybara-service[bot]
Copy link

Switch the default nonsymmetric eigendecomposition implementation to use cusolver.

This regresses a test that checks that eigendecompositions behave well if provided NaNs as input. I think this is the right tradeoff, however, since the cusolver implementation should be faster and the old implementation is still available.

Deprecate the use_magma argument to lax.linalg.eig, replace it with an implementation argument.

This PR builds on an partial PR from @dfm.

Fixes #27265

@copybara-service copybara-service bot force-pushed the test_816815747 branch 3 times, most recently from 92cd12a to c0c8599 Compare October 10, 2025 13:38
…use cusolver.

This regresses a test that checks that eigendecompositions behave well if provided NaNs as input. I think this is the right tradeoff, however, since the cusolver implementation should be faster and the old implementation is still available.

Deprecate the use_magma argument to `lax.linalg.eig`, replace it with an `implementation` argument.

This PR builds on an partial PR from @dfm.

Fixes #27265

PiperOrigin-RevId: 817628185
@copybara-service copybara-service bot merged commit 97b001a into main Oct 10, 2025
@copybara-service copybara-service bot deleted the test_816815747 branch October 10, 2025 14:02
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Feature request: linalg.eig using nvidia cusolver

1 participant