From 46a4d98baa1d43b33717b4377aefeeaf46b9c2ff Mon Sep 17 00:00:00 2001 From: Yong He Date: Mon, 23 Jan 2023 06:59:25 -0800 Subject: Full address insts elimination for backward autodiff. (#2604) Co-authored-by: Yong He --- tests/autodiff/reverse-struct-multi-write.slang | 48 +++++++++++++++++++++++++ 1 file changed, 48 insertions(+) create mode 100644 tests/autodiff/reverse-struct-multi-write.slang (limited to 'tests/autodiff/reverse-struct-multi-write.slang') diff --git a/tests/autodiff/reverse-struct-multi-write.slang b/tests/autodiff/reverse-struct-multi-write.slang new file mode 100644 index 000000000..dd12c7d3d --- /dev/null +++ b/tests/autodiff/reverse-struct-multi-write.slang @@ -0,0 +1,48 @@ + +//TEST(compute):COMPARE_COMPUTE_EX:-slang -compute -shaderobj -output-using-type +//TEST(compute, vulkan):COMPARE_COMPUTE_EX:-vk -compute -shaderobj -output-using-type + +//TEST_INPUT:ubuffer(data=[0 0 0 0 0], stride=4):out,name=outputBuffer +RWStructuredBuffer outputBuffer; + +struct A : IDifferentiable +{ + float x; + float y; +}; + +[BackwardDifferentiable] +A f(A a) +{ + // Read/writes to local struct variables won't be SSA'd out by default. + // The backward diff preparation pass will kick in to create temp vars for them. + A aout; + aout.y = 2 * a.x; + aout.y = aout.y + 2 * a.x; + aout.x = aout.y + 5 * a.x; + + // The result should be equivalent to: + /* + A aout; + var tmp = 2 * a.x; + tmp = tmp + 2 * a.x; + aout.y = tmp; + aout.x = tmp + 5 * a.x; + */ + return aout; + +} + +[numthreads(1, 1, 1)] +void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID) +{ + A a = {1.0, 2.0}; + + var dpa = diffPair(a); + + A.Differential dout = {1.0, 1.0}; + + __bwd_diff(f)(dpa, dout); + outputBuffer[0] = dpa.d.x; // Expect: 13 + outputBuffer[1] = dpa.d.y; // Expect: 0 +} -- cgit v1.2.3