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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
|
// Test calling differentiable function through dynamic dispatch.
//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_INPUT: set g_materials = new StructuredBuffer<MaterialDataBlob>[new MaterialDataBlob{new MaterialHeader{[0, 0, 0, 0]}, new MaterialPayload{[1.0, 1.2, 0.3, 0.5]}}];
RWStructuredBuffer<MaterialDataBlob> g_materials;
public struct ShadingInput
{
public float scale;
}
struct MaterialHeader
{
uint4 header;
};
struct MaterialPayload
{
float4 data;
};
struct MaterialDataBlob
{
MaterialHeader header; // 16B
MaterialPayload payload; // 16B
};
interface IMaterial : IDifferentiable
{
associatedtype MaterialInstance : IMaterialInstance;
[Differentiable]
MaterialInstance setupMaterialInstance( ShadingInput input );
}
interface IMaterialInstance : IDifferentiable
{
[Differentiable]
float eval( float x );
}
[BackwardDerivative(getMaterial_bwd)]
IMaterial getMaterial(int id)
{
return createDynamicObject<IMaterial, MaterialDataBlob>(id, g_materials[id]);
}
void getMaterial_bwd(int id, IDifferentiable d)
{
// Something random
outputBuffer[id] = 2.f;
}
struct Material1: IMaterial
{
typedef MaterialInstance1 MaterialInstance;
MaterialHeader header;
float a;
float b;
float c;
[Differentiable]
MaterialInstance1 setupMaterialInstance( ShadingInput input )
{
MaterialInstance1 instance;
instance.a = a * input.scale;
instance.b = b * input.scale;
instance.c = c * input.scale;
return instance;
}
}
struct MaterialInstance1: IMaterialInstance
{
float a;
float b;
float c;
[Differentiable]
float eval( float x )
{
return a * x * x + b * x + c;
}
}
struct Material2: IMaterial
{
typedef MaterialInstance2 MaterialInstance;
MaterialHeader header;
float a;
float b;
[Differentiable]
MaterialInstance2 setupMaterialInstance( ShadingInput input )
{
MaterialInstance2 instance;
instance.a = a * input.scale * input.scale;
instance.b = b * input.scale * input.scale;
return instance;
}
}
public struct MaterialInstance2: IMaterialInstance
{
float a;
float b;
[Differentiable]
public float eval( float x )
{
return a * x + b;
}
}
[Differentiable]
public float shade(int material, ShadingInput input, float x)
{
IMaterial m = getMaterial(material);
IMaterialInstance mi = m.setupMaterialInstance(input);
return mi.eval(x);
}
//TEST_INPUT: type_conformance Material1:IMaterial = 0
//TEST_INPUT: type_conformance Material2:IMaterial = 1
[shader("compute")]
void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID)
{
outputBuffer[0] = shade(0, {0.5}, 0.6);
// TODO: VERIFY
DifferentialPair<float> dpx = diffPair(3.0);
bwd_diff(shade)(0, {0.5}, dpx, 1.0);
outputBuffer[3] = dpx.d;
}
|
