forked from shader-slang/slang
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathdynamic-dispatch-1.slang
42 lines (35 loc) · 1011 Bytes
/
dynamic-dispatch-1.slang
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
//TEST(compute):COMPARE_COMPUTE:-dx11 -shaderobj
//TEST(compute):COMPARE_COMPUTE:-vk -shaderobj
//TEST(compute):COMPARE_COMPUTE:-cpu -xslang -disable-specialization -shaderobj
//TEST(compute):COMPARE_COMPUTE_EX:-cuda -compute -xslang -disable-specialization -shaderobj
// Test dynamic dispatch code gen for non-static member functions.
[anyValueSize(16)]
interface IInterface
{
int Compute(int inVal);
};
int GenericCompute<T:IInterface>(T obj, int inVal)
{
return obj.Compute(inVal);
}
struct Impl : IInterface
{
int base;
int Compute(int inVal) { return base + inVal * inVal; }
};
int test(int inVal)
{
Impl obj;
obj.base = 1;
return GenericCompute<Impl>(obj, inVal);
}
//TEST_INPUT:ubuffer(data=[0 1 2 3], stride=4):out,name=outputBuffer
RWStructuredBuffer<int> outputBuffer;
[numthreads(4, 1, 1)]
void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID)
{
uint tid = dispatchThreadID.x;
int inVal = outputBuffer[tid];
int outVal = test(inVal);
outputBuffer[tid] = outVal;
}