Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[Feature] Optimize Krylov solver for batched calculations #295

Open
vytautas-a opened this issue Nov 15, 2024 · 1 comment
Open

[Feature] Optimize Krylov solver for batched calculations #295

vytautas-a opened this issue Nov 15, 2024 · 1 comment
Assignees
@vytautas-a
Labels
feature New feature or request

Comments

@vytautas-a
Copy link
Collaborator

Describe the feature

Currently (#294) batching in Krylov solver is implemented straightforwardly using a simple loop. This should be optimized to harness the power of broadcasted tensor operations.

It should be implemented because

No response

Additional context

No response

Would you like to work on this issue?

None
@vytautas-a vytautas-a added the feature New feature or request label Nov 15, 2024
@vytautas-a vytautas-a self-assigned this Nov 15, 2024
@vytautas-a vytautas-a changed the title Optimize Krylov solver for batched calculations [Feature] Optimize Krylov solver for batched calculations Nov 15, 2024
@mlahariya
Copy link
Collaborator

@vytautas-a Are we starting to work on this? Do we also need to align with anton about the krylov solver already implemented?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@vytautas-a vytautas-a

Labels
feature New feature or request
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@mlahariya @vytautas-a