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

add enzyme to AD tests #505

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

test/test_utils.jl Outdated Show resolved Hide resolved
test/test_utils.jl Outdated Show resolved Hide resolved
@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.

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

Successfully merging this pull request may close these issues.

3 participants