diff options
| author | Sai Praveen Bangaru <31557731+saipraveenb25@users.noreply.github.com> | 2025-04-04 14:12:27 -0400 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2025-04-04 18:12:27 +0000 |
| commit | 41e7e565eb3dfa13562cbfa3e8641874c2c6d66c (patch) | |
| tree | f3ba43df8fe2a4c8ca4a63c9e9f577d7e0f7a028 /tests/autodiff/reverse-loop-diff-only-2.slang | |
| parent | 83a42cb76feb1f702ff730040f359cabc01c571a (diff) | |
Add a loop analysis step to infer the exit values of loop phi parameters. (#6696)
* Initial loop analysis pass
* More changes for a single-pass implication propagation
* Update slang-ir-autodiff-loop-analysis.cpp
* Cleanup + new system for loop analysis
* Fixup bugs in loop analysis
* Remove some relation types to simplify the analysis. Add test
* Remove unused
* Address comments
* Fix issue with continue loops
* Update reverse-loop-exit-value-inference-1.slang
* Update reverse-continue-loop.slang
Diffstat (limited to 'tests/autodiff/reverse-loop-diff-only-2.slang')
| -rw-r--r-- | tests/autodiff/reverse-loop-diff-only-2.slang | 15 |
1 files changed, 11 insertions, 4 deletions
diff --git a/tests/autodiff/reverse-loop-diff-only-2.slang b/tests/autodiff/reverse-loop-diff-only-2.slang index 2cc33ecca..cc9e14736 100644 --- a/tests/autodiff/reverse-loop-diff-only-2.slang +++ b/tests/autodiff/reverse-loop-diff-only-2.slang @@ -32,10 +32,17 @@ float infinitesimal(float x) // Test that computeLoop's intermediates have no float sitting // around (must not cache the outvar from 'compute()') -// CHECK: struct s_bwd_prop_computeLoop_Intermediates -// CHECK-NEXT: { -// CHECK-NOT: {{[A-Za-z0-9_]+}} {{[A-Za-z0-9_]+}}[{{.*}}] -// CHECK: } +// +// Further, if loop exit value inference is working correctly, +// then there should be no context type at all. +// +// CHECK-NOT: struct s_bwd_prop_computeLoop_Intermediates +// +// Check that the signature of the s_bwd_prop_computeLoop function only +// contains an inout DiffPair_float_0 and a float. +// +// CHECK: void s_bwd_prop_computeLoop{{[_0-9]*}}(inout DiffPair_float{{[_0-9]*}} dpy{{[_0-9]*}}, float {{[_a-zA-Z0-9]*}}) +// [BackwardDifferentiable] [PreferRecompute] |