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

ENZYME_OPT dbg improvements #147

Open
ZuseZ4 opened this issue Jul 30, 2024 · 0 comments
Open

ENZYME_OPT dbg improvements #147

ZuseZ4 opened this issue Jul 30, 2024 · 0 comments

Comments

@ZuseZ4
Copy link
Member

ZuseZ4 commented Jul 30, 2024

Right now we add extra autodiff wrappers. That works good enough to test if tt are correct, and to debug things where enzyme fails on IR inside the function being differentiated. For some cases like rsmpi Enzyme will however also analyzes which arguments are passed into the function being differentiated and our ENZYME_OPT wrappers loose this information.
We should therefore drop the artificial wrappers and add enzyme_autodiff calls to the same location where we would otherwise add the differentiated functions if we would have used enzyme as a library.
The issue here are our ABI wrappers, which analyze the function that we get back from Enzyme, not sure how much trouble that will cause.

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

No branches or pull requests

1 participant