yum-mirror/slang

Making it easier to work with shaders

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

James Helferty (NVIDIA)render-test: Change D3D12 default to sm_6_5 (#8320)f02b08490

master
914 B26 linesraw
1//TEST_CATEGORY(wave-mask, compute)
2//DISABLE_TEST:COMPARE_COMPUTE_EX:-cpu -compute -shaderobj
3//DISABLE_TEST:COMPARE_COMPUTE_EX:-slang -compute -shaderobj
4//TEST:COMPARE_COMPUTE_EX:-slang -compute -dx12 -profile cs_6_5 -shaderobj -render-feature hardware-device
5//TEST(vulkan):COMPARE_COMPUTE_EX:-vk -compute -shaderobj -render-feature hardware-device
6//TEST:COMPARE_COMPUTE_EX:-cuda -compute -shaderobj
7
8//TEST_INPUT:ubuffer(data=[0 0 0 0 0 0 0 0], stride=4):out,name outputBuffer
9RWStructuredBuffer<int> outputBuffer;
10
11[numthreads(8, 1, 1)]
12void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID)
13{
14    WaveMask mask = 0xff;
15    
16    int idx = int(dispatchThreadID.x);
17    
18    float2 v1 = float2(1, 1 << idx);
19    
20    int r0 = WaveMaskPrefixSum(mask, 1 << idx);
21    float2 r1 = WaveMaskPrefixSum(mask, v1);
22    
23    int r2 = int(r1.x) + int(r1.y) - idx;
24    
25    outputBuffer[idx] = r0 + (r2 << 16);
26}