diff options
Diffstat (limited to 'tests')
4 files changed, 30 insertions, 4 deletions
diff --git a/tests/autodiff-dstdlib/dstdlib-elementary-bwd.slang b/tests/autodiff-dstdlib/dstdlib-elementary-bwd.slang index 379e2c3ef..53972ac2c 100644 --- a/tests/autodiff-dstdlib/dstdlib-elementary-bwd.slang +++ b/tests/autodiff-dstdlib/dstdlib-elementary-bwd.slang @@ -2,7 +2,7 @@ //TEST(compute):COMPARE_COMPUTE_EX:-slang -compute -shaderobj -output-using-type //TEST(compute):COMPARE_COMPUTE_EX:-cpu -compute -output-using-type -shaderobj -//TEST_INPUT:ubuffer(data=[0 0 0 0 0 0], stride=4):out,name=outputBuffer +//TEST_INPUT:ubuffer(data=[0 0 0 0 0 0 0 0], stride=4):out,name=outputBuffer RWStructuredBuffer<float> outputBuffer; typedef DifferentialPair<float> dpfloat; @@ -43,4 +43,14 @@ void computeMain(uint3 dispatchThreadID: SV_DispatchThreadID) __bwd_diff(diffSin)(dpx, 1.0); outputBuffer[4] = dpx.d; // Expect: -1.000000 } + + { + dpfloat dpx = dpfloat(float.getPi() / 3.0, 1.0); + __bwd_diff(sincos)(dpx, 1.0, 0.0); + outputBuffer[5] = dpx.d; // Expect: 0.5 + __bwd_diff(sincos)(dpx, 0.0, 1.0); + outputBuffer[6] = dpx.d; // Expect: -0.8660254 + __bwd_diff(sincos)(dpx, 1.0, 1.0); + outputBuffer[7] = dpx.d; // Expect: -0.3660254 + } } diff --git a/tests/autodiff-dstdlib/dstdlib-elementary-bwd.slang.expected.txt b/tests/autodiff-dstdlib/dstdlib-elementary-bwd.slang.expected.txt index a4b804cb8..17627df68 100644 --- a/tests/autodiff-dstdlib/dstdlib-elementary-bwd.slang.expected.txt +++ b/tests/autodiff-dstdlib/dstdlib-elementary-bwd.slang.expected.txt @@ -3,4 +3,7 @@ type: float 7.389056 0.000000 1.000000 --1.00000
\ No newline at end of file +-1.000000 +0.500000 +-0.866025 +-0.366025 diff --git a/tests/autodiff-dstdlib/dstdlib-sqrt.slang b/tests/autodiff-dstdlib/dstdlib-sqrt.slang index 15573c4ef..d68a2697c 100644 --- a/tests/autodiff-dstdlib/dstdlib-sqrt.slang +++ b/tests/autodiff-dstdlib/dstdlib-sqrt.slang @@ -1,7 +1,7 @@ //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 0 0 0 0], stride=4):out,name=outputBuffer +//TEST_INPUT:ubuffer(data=[0 0 0 0 0 0 0 0 0 0 0 0], stride=4):out,name=outputBuffer RWStructuredBuffer<float> outputBuffer; typedef DifferentialPair<float> dpfloat; @@ -50,4 +50,13 @@ void computeMain(uint3 dispatchThreadID: SV_DispatchThreadID) outputBuffer[6] = dpx.d[0]; // Expect: 0.158114 outputBuffer[7] = dpx.d[1]; // Expect: 0.577350 } + + { + var dpx = diffPair(float2x2(4.0, 9.0, 16.0, 25.0), float2x2(0.0, 0.0, 0.0, 0.0)); + __bwd_diff(sqrt)(dpx, float2x2(1.0, 2.0, 3.0, 4.0)); + outputBuffer[8] = dpx.d[0][0]; // Expect: 0.25 + outputBuffer[9] = dpx.d[0][1]; // Expect: 0.3333 + outputBuffer[10] = dpx.d[1][0]; // Expect: 0.375 + outputBuffer[11] = dpx.d[1][1]; // Expect: 0.4 + } } diff --git a/tests/autodiff-dstdlib/dstdlib-sqrt.slang.expected.txt b/tests/autodiff-dstdlib/dstdlib-sqrt.slang.expected.txt index fe6487fef..7e0fdf02f 100644 --- a/tests/autodiff-dstdlib/dstdlib-sqrt.slang.expected.txt +++ b/tests/autodiff-dstdlib/dstdlib-sqrt.slang.expected.txt @@ -6,4 +6,8 @@ type: float 0.000000 0.000000 0.158114 -0.577350
\ No newline at end of file +0.577350 +0.250000 +0.333333 +0.375000 +0.400000
\ No newline at end of file |