diff options
| author | Yong He <yonghe@outlook.com> | 2023-01-10 12:42:55 -0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-01-10 12:42:55 -0800 |
| commit | 2f422087ed04940f6b6b351605e61d48ce1989ce (patch) | |
| tree | 522f8027173732d903a906081238b12863d73fb8 /tests | |
| parent | eb813fbd8750ed1ab66d73f5fa29ae8f2407e8af (diff) | |
Nested bwd-diff func call context save/restore. (#2584)
Co-authored-by: Yong He <yhe@nvidia.com>
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/autodiff/reverse-nested-calls.slang | 5 | ||||
| -rw-r--r-- | tests/autodiff/reverse-nested-calls.slang.expected.txt | 2 |
2 files changed, 4 insertions, 3 deletions
diff --git a/tests/autodiff/reverse-nested-calls.slang b/tests/autodiff/reverse-nested-calls.slang index 2b55efd60..55d22e191 100644 --- a/tests/autodiff/reverse-nested-calls.slang +++ b/tests/autodiff/reverse-nested-calls.slang @@ -10,7 +10,8 @@ typedef float.Differential dfloat; [BackwardDifferentiable] float g(float y) { - return 4.0f * y; + float result = 2.0f * y; + return result * result; } [BackwardDifferentiable] @@ -25,5 +26,5 @@ void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID) dpfloat dpa = dpfloat(1.0, 0.0); __bwd_diff(f)(dpa, 1.0f); - outputBuffer[0] = dpa.d; // Expect: 24.0 + outputBuffer[0] = dpa.d; // Expect: 96.0 } diff --git a/tests/autodiff/reverse-nested-calls.slang.expected.txt b/tests/autodiff/reverse-nested-calls.slang.expected.txt index 0a39c4da6..94b18f3fe 100644 --- a/tests/autodiff/reverse-nested-calls.slang.expected.txt +++ b/tests/autodiff/reverse-nested-calls.slang.expected.txt @@ -1,5 +1,5 @@ type: float -24.000000 +96.000000 0.000000 0.000000 0.000000 |