Update Enzyme usage in test files #162
Merged
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Replace all instances of `AutoEnzyme()` with `AutoEnzyme(; mode=Enzyme.set_runtime_activity(Enzyme.Reverse), function_annotation=Enzyme.Const )` in the following files: * `test/inference/repgradelbo_distributionsad.jl` * `test/inference/repgradelbo_locationscale_bijectors.jl` * `test/inference/repgradelbo_locationscale.jl` * `test/inference/scoregradelbo_distributionsad.jl` * `test/inference/scoregradelbo_locationscale_bijectors.jl`
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
![github-actions[bot]](https://avatars.githubusercontent.com/in/15368?s=60&v=4){kind=small}
There was a problem hiding this comment.
Benchmark Results
| Benchmark suite | Current: acf55fc | Previous: 3b795fd | Ratio |
|---|---|---|---|
normal/RepGradELBO + STL/meanfield/Zygote |
15198724796 ns |
16473872136 ns |
0.92 |
normal/RepGradELBO + STL/meanfield/ForwardDiff |
3240455962.5 ns |
3304125733.5 ns |
0.98 |
normal/RepGradELBO + STL/meanfield/ReverseDiff |
3069781398 ns |
3094818311.5 ns |
0.99 |
normal/RepGradELBO + STL/fullrank/Zygote |
14849629285 ns |
16338470403 ns |
0.91 |
normal/RepGradELBO + STL/fullrank/ForwardDiff |
3567893647.5 ns |
3688626107 ns |
0.97 |
normal/RepGradELBO + STL/fullrank/ReverseDiff |
5663062388 ns |
5646882456 ns |
1.00 |
normal/RepGradELBO/meanfield/Zygote |
7039209760 ns |
7530702671 ns |
0.93 |
normal/RepGradELBO/meanfield/ForwardDiff |
2432843392 ns |
2353602785 ns |
1.03 |
normal/RepGradELBO/meanfield/ReverseDiff |
1324250921 ns |
1313395716.5 ns |
1.01 |
normal/RepGradELBO/fullrank/Zygote |
7055155643 ns |
7682697941 ns |
0.92 |
normal/RepGradELBO/fullrank/ForwardDiff |
2633336787 ns |
2635490263.5 ns |
1.00 |
normal/RepGradELBO/fullrank/ReverseDiff |
2446923916 ns |
2436281408 ns |
1.00 |
normal + bijector/RepGradELBO + STL/meanfield/Zygote |
23812917423 ns |
25565717237 ns |
0.93 |
normal + bijector/RepGradELBO + STL/meanfield/ForwardDiff |
9910518313 ns |
10309343356 ns |
0.96 |
normal + bijector/RepGradELBO + STL/meanfield/ReverseDiff |
4461054562 ns |
4476042343 ns |
1.00 |
normal + bijector/RepGradELBO + STL/fullrank/Zygote |
24066961876 ns |
25818366598 ns |
0.93 |
normal + bijector/RepGradELBO + STL/fullrank/ForwardDiff |
10881089019 ns |
10786917463 ns |
1.01 |
normal + bijector/RepGradELBO + STL/fullrank/ReverseDiff |
7750916721 ns |
7641622123 ns |
1.01 |
normal + bijector/RepGradELBO/meanfield/Zygote |
15331386356 ns |
15735398805 ns |
0.97 |
normal + bijector/RepGradELBO/meanfield/ForwardDiff |
9388134243 ns |
9335421424 ns |
1.01 |
normal + bijector/RepGradELBO/meanfield/ReverseDiff |
2544179631 ns |
2541208289 ns |
1.00 |
normal + bijector/RepGradELBO/fullrank/Zygote |
15668964233 ns |
16236341205 ns |
0.97 |
normal + bijector/RepGradELBO/fullrank/ForwardDiff |
9871406635 ns |
10048414269 ns |
0.98 |
normal + bijector/RepGradELBO/fullrank/ReverseDiff |
4050689049.5 ns |
3948466684 ns |
1.03 |
This comment was automatically generated by workflow using github-action-benchmark.
Codecov ReportAll modified and coverable lines are covered by tests ✅
Additional details and impacted files@@ Coverage Diff @@
## main #162 +/- ##
=======================================
Coverage ? 91.73%
=======================================
Files ? 13
Lines ? 351
Branches ? 0
=======================================
Hits ? 322
Misses ? 29
Partials ? 0 ☔ View full report in Codecov by Sentry. 🚀 New features to boost your workflow:
|
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
1 participant
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Replace all instances of
AutoEnzyme()withAutoEnzyme(; mode=Enzyme.set_runtime_activity(Enzyme.Reverse), function_annotation=Enzyme.Const )in the following files:test/inference/repgradelbo_distributionsad.jltest/inference/repgradelbo_locationscale_bijectors.jltest/inference/repgradelbo_locationscale.jltest/inference/scoregradelbo_distributionsad.jltest/inference/scoregradelbo_locationscale_bijectors.jl