From 2f422087ed04940f6b6b351605e61d48ce1989ce Mon Sep 17 00:00:00 2001 From: Yong He Date: Tue, 10 Jan 2023 12:42:55 -0800 Subject: Nested bwd-diff func call context save/restore. (#2584) Co-authored-by: Yong He --- tests/autodiff/reverse-nested-calls.slang | 5 +++-- tests/autodiff/reverse-nested-calls.slang.expected.txt | 2 +- 2 files changed, 4 insertions(+), 3 deletions(-) (limited to 'tests') 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 -- cgit v1.2.3