tests/language-feature/generics/parameter-block-unify.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

//TEST(compute):COMPARE_COMPUTE_EX(filecheck-buffer=CHECK):-slang -compute -shaderobj -output-using-type
//TEST(compute, vulkan):COMPARE_COMPUTE_EX(filecheck-buffer=CHECK):-vk -compute -shaderobj -output-using-type
//TEST(compute, vulkan):COMPARE_COMPUTE_EX(filecheck-buffer=CHECK):-mtl -compute -shaderobj -output-using-type -render-features argument-buffer-tier-2

struct TestStruct<Format:__BuiltinIntegerType, let count : int>
{
    Format f;
};

Format testFunction<Format : __BuiltinIntegerType, let count : int>(TestStruct<Format, count> data)
{
    return data.f + __int_cast<Format>(count);
}

//TEST_INPUT: set testBlock = new TestStruct<int, 12>{1}
ParameterBlock<TestStruct<int, 12>> testBlock;

//TEST_INPUT: set testBlock2 = new TestStruct<int, 12>{2}
ConstantBuffer<TestStruct<int, 2>> testBlock2;

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

[numthreads(1, 1, 1)]
void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID)
{
    // CHECK: 13
    outputBuffer[0] = testFunction(testBlock);
    // CHECK: 13
    outputBuffer[1] = testFunction<int, 12>(testBlock);
    // CHECK: 4
    outputBuffer[2] = testFunction(testBlock2);
}