blob: e8ffea6255259392fd54ce9e4d78411c5f41cf8f (
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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
|
//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;
[Differentiable] [PreferRecompute]
float3 fetch(float2 uv)
{
if (uv.x > 0.5f)
{
if (uv.x > 0.7f)
return float3(2.) * uv.y;
else
return float3(1.) * uv.y;
}
else
{
if (uv.x > 0.3f)
return float3(4.) * uv.y;
else
return float3(3.) * uv.y;
}
}
[numthreads(1, 1, 1)]
void computeMain(uint3 dispatchThreadID: SV_DispatchThreadID)
{
float2 uv = (float2)dispatchThreadID.xy / float2(512, 512);
float3 color = fetch(uv);
outputBuffer[0] = color.x; // Expect: 0.0
{
DifferentialPair<float2> dpuv = diffPair(float2(0.6f));
bwd_diff(fetch)(dpuv, float3(1.f));
outputBuffer[1] = dpuv.d.y; // Expect: 1.0 * 3 = 3
}
{
DifferentialPair<float2> dpuv = diffPair(float2(0.8f));
bwd_diff(fetch)(dpuv, float3(1.f));
outputBuffer[2] = dpuv.d.y; // Expect: 2.0 * 3 = 6
}
{
DifferentialPair<float2> dpuv = diffPair(float2(0.1f));
bwd_diff(fetch)(dpuv, float3(1.f));
outputBuffer[3] = dpuv.d.y; // Expect: 3.0 * 3 = 9
}
{
DifferentialPair<float2> dpuv = diffPair(float2(0.4f));
bwd_diff(fetch)(dpuv, float3(1.f));
outputBuffer[4] = dpuv.d.y; // Expect: 4.0 * 3 = 12
}
}
|
