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.

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

Vector / chunked / batch mode #312

Closed
gdalle opened this issue Jun 6, 2024 · 4 comments · Fixed by #499
Closed

Vector / chunked / batch mode #312

gdalle opened this issue Jun 6, 2024 · 4 comments · Fixed by #499
Labels
backend Related to one or more autodiff backends core Related to the core utilities of the package

Comments

@gdalle
Copy link
Member

gdalle commented Jun 6, 2024

This one is rather easy in principle: just wrap the JVP/VJP/HVP seed in a Batch struct that is essentially a named tuple, and dispatch on that

@gdalle gdalle added backend Related to one or more autodiff backends core Related to the core utilities of the package labels Jun 8, 2024
@gdalle
Copy link
Member Author

gdalle commented Jun 20, 2024

Partially solved by #320, still needs:

@gdalle
Copy link
Member Author

gdalle commented Jul 16, 2024

It would be nice to avoid the distinction between operator and operator_batched, but that would require wrapping inputs into a Single struct or so

@gdalle
Copy link
Member Author

gdalle commented Aug 29, 2024

Further improved by #412, which makes batch mode the default that extensions need to implement

@gdalle
Copy link
Member Author

gdalle commented Sep 23, 2024

Batch mode for Enzyme is nearly complete since #471, it only needs EnzymeAD/Enzyme.jl#1883 to work with out-of-place functions

@gdalle gdalle linked a pull request Sep 25, 2024 that will close this issue
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
backend Related to one or more autodiff backends core Related to the core utilities of the package
Projects
None yet
Development

Successfully merging a pull request may close this issue.

1 participant