diff options
Diffstat (limited to 'tests/autodiff')
4 files changed, 84 insertions, 0 deletions
diff --git a/tests/autodiff/self-differential-generic-type-synthesis.slang b/tests/autodiff/self-differential-generic-type-synthesis.slang new file mode 100644 index 000000000..8d225dec2 --- /dev/null +++ b/tests/autodiff/self-differential-generic-type-synthesis.slang @@ -0,0 +1,36 @@ +//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<float> outputBuffer; + +// Test that struct types made up of differentiable members who are self-differential (i.e. their Differential type is the same as their type) +// are considered self-differential as well. We should be able to assign T.Differential = T and T = T.Differential without errors. +// + + +struct Ray<let N: int> : IDifferentiable { + float a; + vector<float, N> dir, o; +} + +[numthreads(1, 1, 1)] +void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID) +{ + Ray<4> ray = Ray<4>(); + Ray<4>.Differential ray2; + + ray.a = 1.f; + ray.o = float4(3.f, 4.f, 2.5f, 1.f); + + ray2 = ray; + + float t = 0.f; + float.Differential dt = 0.f; + + t = dt; + + outputBuffer[0] = t; + outputBuffer[1] = ray2.o.y; + outputBuffer[2] = Ray<4>.dadd(ray2, ray2).o.w; +} diff --git a/tests/autodiff/self-differential-generic-type-synthesis.slang.expected.txt b/tests/autodiff/self-differential-generic-type-synthesis.slang.expected.txt new file mode 100644 index 000000000..e3160fd7f --- /dev/null +++ b/tests/autodiff/self-differential-generic-type-synthesis.slang.expected.txt @@ -0,0 +1,6 @@ +type: float +0.000000 +4.000000 +2.000000 +0.000000 +0.000000 diff --git a/tests/autodiff/self-differential-type-synthesis.slang b/tests/autodiff/self-differential-type-synthesis.slang new file mode 100644 index 000000000..7f95891c6 --- /dev/null +++ b/tests/autodiff/self-differential-type-synthesis.slang @@ -0,0 +1,36 @@ +//TEST(compute):COMPARE_COMPUTE_EX:-slang -compute -shaderobj -output-using-type +//TEST(compute, vulkan):COMPARE_COMPUTE_EX:-vk -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], stride=4):out,name=outputBuffer +RWStructuredBuffer<float> outputBuffer; + +// Test that struct types made up of differentiable members who are self-differential (i.e. their Differential type is the same as their type) +// are considered self-differential as well. We should be able to assign T.Differential = T and T = T.Differential without errors. +// 1 + +struct Ray : IDifferentiable { + float a; + float3 dir, o; +} + +[numthreads(1, 1, 1)] +void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID) +{ + Ray ray = Ray(); + Ray.Differential ray2; + + ray.a = 1.f; + ray.o = float3(3.f, 4.f, 2.5f); + + ray2 = ray; + + float t = 0.f; + float.Differential dt = 0.f; + + t = dt; + + outputBuffer[0] = t; + outputBuffer[1] = ray2.o.y; + outputBuffer[2] = Ray.dadd(ray2, ray2).a; +} diff --git a/tests/autodiff/self-differential-type-synthesis.slang.expected.txt b/tests/autodiff/self-differential-type-synthesis.slang.expected.txt new file mode 100644 index 000000000..e3160fd7f --- /dev/null +++ b/tests/autodiff/self-differential-type-synthesis.slang.expected.txt @@ -0,0 +1,6 @@ +type: float +0.000000 +4.000000 +2.000000 +0.000000 +0.000000 |