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
|
//TEST_CATEGORY(wave, compute)
//DISABLE_TEST:COMPARE_COMPUTE_EX:-cpu -compute -shaderobj
//DISABLE_TEST:COMPARE_COMPUTE_EX:-slang -compute -shaderobj
//TEST:COMPARE_COMPUTE_EX:-slang -compute -dx12 -profile sm_6_5 -shaderobj
//TEST:COMPARE_COMPUTE_EX:-vk -compute -shaderobj
//TEST:COMPARE_COMPUTE_EX:-cuda -compute -capability cuda_sm_7_0 -shaderobj
//TEST_INPUT:ubuffer(data=[0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0], stride=4):out,name outputBuffer
RWStructuredBuffer<uint> outputBuffer;
[numthreads(8, 1, 1)]
[shader("compute")]
void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID)
{
uint index = int(dispatchThreadID.x);
// Split into two groups.
uint4 mask = 0b00001111;
if (index >= 4)
{
mask = 0b11110000;
}
//
// WaveMultiPrefixSum.
// Results in hex: [0 1 3 7], [0 10 30 70]
//
uint sumValue = WaveMultiPrefixSum(1 << index, mask);
const uint sumBaseIndex = 0;
outputBuffer[sumBaseIndex + index] = sumValue;
//
// WaveMultiPrefixProduct.
// Results in hex: [1 1 2 8], [1 10 200 8000]
//
uint productValue = WaveMultiPrefixProduct(1 << index, mask);
const uint productBaseIndex = 8;
outputBuffer[productBaseIndex + index] = productValue;
//
// WaveMultiPrefixBitAnd.
// This prefix operation starts with all bits set.
// Results in hex: [FFFFFFFF 1 1 1], [FFFFFFFF F F F]
//
uint andBits = 0b1;
if (index >= 4)
{
andBits = 0b1111;
}
uint andValue = WaveMultiPrefixBitAnd(andBits, mask);
const uint andBaseIndex = 16;
outputBuffer[andBaseIndex + index] = andValue;
//
// WaveMultiPrefixBitOr.
// Results in hex: [0 1 3 7], [0 10 30 70]
//
uint orValue = WaveMultiPrefixBitOr(1 << index, mask);
const uint orBaseIndex = 24;
outputBuffer[orBaseIndex + index] = orValue;
//
// WaveMultiPrefixBitXor.
// Results in hex: [0 1 3 7], [0 F 0 F]
//
uint xorBits = (1 << index);
if (index >= 4)
{
xorBits = 0b1111;
}
uint xorValue = WaveMultiPrefixBitXor(xorBits, mask);
const uint xorBaseIndex = 32;
outputBuffer[xorBaseIndex + index] = xorValue;
}
|
