How to compute the JVP for multiple-valued outputs? #2321
Unanswered
RohanGautam
asked this question in
Q&A
Replies: 1 comment
-
I am not aware of support for it, so you'll likely need to write a wrapper around it which also accepts the output vector by reference. |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
Uh oh!
There was an error while loading. Please reload this page.
-
I have a c++ function with a
std::vec
as input and a (larger)std::vec
for it's output. I couldn't find a clear way to compute the JVP of the output in this multi-input, multi-output case. Seems like it should be possible (the JVP is computed in the Enzyme-JAX interop library example), but I couldn't find any way to do this from the documentation.A sample fuction signature:
Any pointers to relavant resources would be appreciated!
Beta Was this translation helpful? Give feedback.
All reactions