yum-mirror/slang

Making it easier to work with shaders

git clone https://git.yummers.dev/yum-mirror/slang

Darren WihandiAdd CoopVec load/store pointer overloads (#6822)d108bfa67

master
908 B32 linesraw
1//TEST(compute):COMPARE_COMPUTE(filecheck-buffer=CHECK):-vk -render-feature cooperative-vector -emit-spirv-directly
2
3//TEST_INPUT: set inputBuffer = ubuffer(data=[1 2 3 4 5 6 7 8 9 10 11 12], stride=4);
4uniform int32_t* inputBuffer;
5
6//TEST_INPUT: set outputBuffer = out ubuffer(data=[0 0 0 0 0 0 0 0], stride=4);
7uniform int32_t* outputBuffer;
8
9// CHECK: 9
10// CHECK-NEXT: A
11// CHECK-NEXT: B
12// CHECK-NEXT: C
13// CHECK-NEXT: 1
14// CHECK-NEXT: 2
15// CHECK-NEXT: 3
16// CHECK-NEXT: 4
17
18[shader("compute")]
19[numthreads(1, 1, 1)]
20void computeMain()
21{
22    // First half of input.
23    let a = coopVecLoad<4, int32_t>(inputBuffer, 0);
24    // Second half of input.
25    let b = coopVecLoad<4, int32_t>(inputBuffer + 4, 4*4);
26
27    // Store second half of input to first half of output buffer.
28    b.store(outputBuffer, 0);
29    // Store first half of input to second half of output buffer.
30    a.store(outputBuffer, 4*4);
31}
32