blob: fb28702021ddf96f1e04166c1da62793995c444f (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
|
//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:-cuda -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>(0.0, float4(0.0), float4(0.0));;
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;
}
|
