yum-mirror/slang
Making it easier to work with shaders
git clone https://git.yummers.dev/yum-mirror/slang
d108bfa67
master
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