tests/spirv/array-uniform-param.slang


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

// array-uniform-param.slang

//TESTD:SIMPLE:-target spirv -entry computeMain -stage compute -emit-spirv-directly -force-glsl-scalar-layout
//TEST(compute, vulkan):COMPARE_COMPUTE_EX(filecheck-buffer=BUFFER):-vk -compute -output-using-type

// Test direct SPIR-V emit on arrays in uniforms.

//TEST_INPUT:set resultBuffer = out ubuffer(data=[0 0 0 0], stride=4)
RWStructuredBuffer<uint> resultBuffer;

struct Data
{
    uint arr[4];
}

struct Data2
{
    uint arr[4u];
}

//TEST_INPUT: set Uniforms.d = new Data{{1,2,3,4}};
//TEST_INPUT: set Uniforms.d2 = new Data2{{1,2,3,4}};

cbuffer Uniforms
{
    Data d;
    Data2 d2;
}

// BUFFER: 3

[numthreads(4,1,1)]
void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID)
{
    uint threadId = dispatchThreadID.x;
    uint result = d.arr[1];
    var u2 = reinterpret<Data>(d2);
    result += u2.arr[0];
    resultBuffer[threadId] = result;
}