summaryrefslogtreecommitdiffstats
path: root/tests/language-feature/if-let/if-let-1.slang
blob: c197f26eaf33839cfc26e622d033b5ff64763108 (plain)
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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
//TEST(compute):COMPARE_COMPUTE(filecheck-buffer=CHECK): -cpu -compute -shaderobj
//TEST(compute):COMPARE_COMPUTE(filecheck-buffer=CHECK): -cuda -compute -shaderobj
//TEST(compute, vulkan):COMPARE_COMPUTE_EX(filecheck-buffer=CHECK):-vk -compute -shaderobj

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


interface IFoo
{
    int foo(int a);
}

struct MyImpl1 : IFoo
{
    int foo(int a) { return a; }
}

struct MyImpl2 : IFoo
{
    int foo(int a) { return a + 5; }
}

int test(IFoo foo, int idx)
{
    int val = 0;
    if (let a = foo as MyImpl1)
    {
        val = a.foo(idx);
    }
    else if (let a = foo as MyImpl2)
    {
        val = a.foo(idx);
    }
    return (val);
}

int test1<T>(T t)
{
    if (let a = t as uint)
    {
        return 1;
    }
    else if(let a = t as float)
    {
        return 2;
    }
    else if (let a = t as double)
    {
        return 3;
    }
    else if (let a = t as int)
    {
        return 4;
    }
    else if (let a = t as uint64_t)
    {
        return 5;
    }
    else
    {
        return 6;
    }
}


[numthreads(1, 1, 1)]
void computeMain(uint3 dispatchThreadID: SV_DispatchThreadID)
{
    MyImpl1 impl1;
    MyImpl2 impl2;
    // CHECK: 1
    // CHECK: 7
    outputBuffer[0] = test(impl1, 1);
    outputBuffer[1] = test(impl2, 2);

    // CHECK: 1
    outputBuffer[2] = test1(2U);
    // CHECK: 2
    outputBuffer[3] = test1(2.0f);
    // CHECK: 3
    outputBuffer[4] = test1(2.0lf);
    // CHECK: 4
    outputBuffer[5] = test1(2);
    // CHECK: 5
    outputBuffer[6] = test1(2LLU);
    // CHECK: 6
    outputBuffer[7] = test1(impl1);
}