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 ++++++++++++++++++++++ .../reverse-struct-multi-write.slang.expected.txt | 6 +++ 2 files changed, 54 insertions(+) create mode 100644 tests/autodiff/reverse-struct-multi-write.slang create mode 100644 tests/autodiff/reverse-struct-multi-write.slang.expected.txt (limited to 'tests/autodiff') 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 +} diff --git a/tests/autodiff/reverse-struct-multi-write.slang.expected.txt b/tests/autodiff/reverse-struct-multi-write.slang.expected.txt new file mode 100644 index 000000000..403f2ffd4 --- /dev/null +++ b/tests/autodiff/reverse-struct-multi-write.slang.expected.txt @@ -0,0 +1,6 @@ +type: float +13.000000 +0.000000 +0.000000 +0.000000 +0.000000 -- cgit v1.2.3