From 259a015feb9d4ab65e8fbba32f6c777e92780cc7 Mon Sep 17 00:00:00 2001 From: Yong He Date: Wed, 22 Mar 2023 21:16:35 -0700 Subject: Type legalization and autodiff bug fixes. (#2722) * Bug fixes. * Fix. * Only perform autodiff for functions whose derivative is actually used. * Fix loop optimize bug. * Fix high order diff. * Fix trivial diff func generation. * Fixes. * Cleanup. --------- Co-authored-by: Yong He --- tests/autodiff-dstdlib/determinant.slang | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) create mode 100644 tests/autodiff-dstdlib/determinant.slang (limited to 'tests/autodiff-dstdlib/determinant.slang') diff --git a/tests/autodiff-dstdlib/determinant.slang b/tests/autodiff-dstdlib/determinant.slang new file mode 100644 index 000000000..d2e699551 --- /dev/null +++ b/tests/autodiff-dstdlib/determinant.slang @@ -0,0 +1,24 @@ +//TEST(compute, vulkan):COMPARE_COMPUTE_EX:-vk -compute -shaderobj -output-using-type +//TEST(compute):COMPARE_COMPUTE_EX:-slang -compute -shaderobj -output-using-type + +//TEST_INPUT:ubuffer(data=[0 0 0 0], stride=4):out,name=outputBuffer +RWStructuredBuffer outputBuffer; + +[BackwardDifferentiable] +float diffDeterminant(float2x2 x) +{ + return determinant(x); +} + +[numthreads(1, 1, 1)] +void computeMain(uint3 dispatchThreadID: SV_DispatchThreadID) +{ + { + var dpx = diffPair(float2x2(1.0, 2.0, 3.0, 4.0)); + __bwd_diff(diffDeterminant)(dpx, 1.0); + outputBuffer[0] = dpx.d[0][0]; + outputBuffer[1] = dpx.d[0][1]; + outputBuffer[2] = dpx.d[1][0]; + outputBuffer[3] = dpx.d[1][1]; + } +} -- cgit v1.2.3