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
|
//TEST(compute):COMPARE_COMPUTE(filecheck-buffer=CHECK):-vk -compile-arg -skip-spirv-validation -emit-spirv-directly
//TEST(compute):COMPARE_COMPUTE(filecheck-buffer=CHECK):-dx12
//TEST(compute):COMPARE_COMPUTE(filecheck-buffer=CHECK):-mtl
//TEST(compute):COMPARE_COMPUTE(filecheck-buffer=CHECK):-cpu
//TEST(compute):COMPARE_COMPUTE(filecheck-buffer=CHECK):-cuda
//TEST(compute):COMPARE_COMPUTE(filecheck-buffer=CHECK):-wgpu
// CHECK: 1
// CHECK-NEXT: 2
// CHECK-NEXT: 3
// CHECK-NEXT: 4
// CHECK-NEXT: 5
// CHECK-NEXT: 6
// CHECK-NEXT: 7
// CHECK-NEXT: 8
// CHECK-NEXT: 21
// CHECK-NEXT: 7A
// CHECK-NEXT: FFFFFFFA
// CHECK-NEXT: A
// CHECK-NEXT: 67
// CHECK-NEXT: FFFFFFEF
// CHECK-NEXT: 32
// CHECK-NEXT: FFFFFFA9
//TEST_INPUT:ubuffer(data=[-1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1], stride=4):out,name=i32Buffer
RWStructuredBuffer<int> i32Buffer;
[numthreads(1, 1, 1)]
void computeMain()
{
// 32-bit tests
{
// Simple hex extraction to test, varying the offset.
uint value = 0x87654321;
i32Buffer[0] = bitfieldExtract(value, 4 * 0, 4);
i32Buffer[1] = bitfieldExtract(value, 4 * 1, 4);
i32Buffer[2] = bitfieldExtract(value, 4 * 2, 4);
i32Buffer[3] = bitfieldExtract(value, 4 * 3, 4);
i32Buffer[4] = bitfieldExtract(value, 4 * 4, 4);
i32Buffer[5] = bitfieldExtract(value, 4 * 5, 4);
i32Buffer[6] = bitfieldExtract(value, 4 * 6, 4);
i32Buffer[7] = bitfieldExtract(value, 4 * 7, 4);
// Now varying the bit length
value = 0b00111011111011110001111010100001;
i32Buffer[8] = bitfieldExtract(value, 0, 6);
i32Buffer[9] = bitfieldExtract(value, 6, 8);
// Sign extension case
// - For unsigned data types, the most significant bits of the result will be set to zero.
// - For signed data types, the most significant bits will be set to the value of bit offset + base - 1
// (i.e., it is sign extended to the width of the return type).
i32Buffer[10] = bitfieldExtract(0b1010111, 3, 4); // 0b1010 -> 0b11111111111111111111111111111010
i32Buffer[11] = bitfieldExtract(0b1010111u, 3, 4); // 0b1111 -> 0b00000000000000000000000000001010
// Component-wise extraction
int4 val4 = int4(0x12345678, 0x9abcdef0, 0x87654321, 0xfedcba98);
int4 ext4 = bitfieldExtract(val4, 4, 8);
i32Buffer[12] = ext4.x;
i32Buffer[13] = ext4.y;
i32Buffer[14] = ext4.z;
i32Buffer[15] = ext4.w;
}
}
|
