-
Notifications
You must be signed in to change notification settings - Fork 36
add enzyme to AD tests #505
New issue
Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? # to your account
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