Skip to content

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

Draft
wants to merge 6 commits into
base: master
Choose a base branch
from
Draft

add enzyme to AD tests #505

wants to merge 6 commits into from

Conversation

st--
Copy link
Member

@st-- st-- commented May 2, 2023

Summary

Proposed changes

  • ...

What alternatives have you considered?

Breaking changes

@codecov
Copy link

codecov bot commented May 2, 2023

Codecov Report

Patch coverage has no change and project coverage change: -62.85 ⚠️

Comparison is base (ef6d459) 94.16% compared to head (858dd66) 31.31%.

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     

see 44 files with indirect coverage changes

☔ View full report in Codecov by Sentry.
📢 Do you have feedback about the report comment? Let us know in this issue.

st-- and others added 5 commits May 2, 2023 20:10
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>
# 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))
Copy link
Member

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...

@simsurace
Copy link
Member

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.

# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants