tests/language-feature/generics/variadic-partial-inference.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

//TEST(compute):COMPARE_COMPUTE(filecheck-buffer=CHECK): -shaderobj -output-using-type
//TEST(compute):COMPARE_COMPUTE(filecheck-buffer=CHECK): -cpu -shaderobj -output-using-type
//TEST(compute):COMPARE_COMPUTE(filecheck-buffer=CHECK): -vk -shaderobj -output-using-type

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

void aggregate(inout int v, int u)
{
    v = v * u;
}

T test<T:IInteger, each U : IInteger>(T t, expand each U u)
{
    int v = 1;
    expand aggregate(v, (each u).toInt());
    return t + T(v);
}

[numthreads(1, 1, 1)]
void computeMain()
{
    // CHECK: 11
    outputBuffer[0] = test<int>(5, 1, 2, 3);
}