-
Notifications
You must be signed in to change notification settings - Fork 34
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
add enzyme to AD tests #505
base: master
Are you sure you want to change the base?
Conversation
Codecov ReportPatch coverage has no change and project coverage change:
Additional details and impacted files@@ Coverage Diff @@
## master #505 +/- ##
===========================================
- Coverage 94.16% 31.31% -62.85%
===========================================
Files 52 52
Lines 1387 1373 -14
===========================================
- Hits 1306 430 -876
- Misses 81 943 +862 ☔ View full report in Codecov by Sentry. |
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
…elFunctions.jl into st/enzyme_ad
# f_prime(flatargs) = f(reshape(flatargs, shape...)) | ||
# return Enzyme.gradient(Enzyme.Reverse, f_prime, reshape(args, prod(shape))) | ||
d_args = zero(args) | ||
Enzyme.autodiff(Enzyme.Reverse, f, Enzyme.Active, Enzyme.Duplicated(args, d_args)) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I don't think you need the Enzyme.Active
but not sure...
I think it's worth re-running these tests. I just did a bunch of tests myself (not this PR though) and most kernels seem to work now. |
Summary
Proposed changes
What alternatives have you considered?
Breaking changes