xref: /llvm-project/llvm/test/CodeGen/SPIRV/structurizer/cf.switch.opswitch.simple.ll (revision 1ed65febd996eaa018164e880c87a9e9afc6f68d)
1; RUN: llc -mtriple=spirv-unknown-vulkan-compute -O0 %s -o - | FileCheck %s
2; RUN: %if spirv-tools %{ llc -O0 -mtriple=spirv-unknown-vulkan-compute %s -o - -filetype=obj | spirv-val %}
3
4;
5; int foo() { return 200; }
6;
7; [numthreads(1, 1, 1)]
8; void main() {
9;   int result;
10;
11;   int a = 0;
12;   switch(a) {
13;     case -3:
14;       result = -300;
15;       break;
16;     case 0:
17;       result = 0;
18;       break;
19;     case 1:
20;       result = 100;
21;       break;
22;     case 2:
23;       result = foo();
24;       break;
25;     default:
26;       result = 777;
27;       break;
28;   }
29;
30;   switch(int c = a) {
31;     case -4:
32;       result = -400;
33;       break;
34;     case 4:
35;       result = 400;
36;       break;
37;   }
38; }
39
40; CHECK: %[[#func_14:]] = OpFunction %[[#uint:]] DontInline %[[#]]
41; CHECK:    %[[#bb25:]] = OpLabel
42; CHECK:                  OpReturnValue %[[#]]
43; CHECK:                  OpFunctionEnd
44; CHECK: %[[#func_15:]] = OpFunction %[[#void:]] DontInline %[[#]]
45; CHECK:    %[[#bb26:]] = OpLabel
46; CHECK:                  OpSelectionMerge %[[#bb27:]] None
47; CHECK:                  OpSwitch %[[#]] %[[#bb28:]] 4294967293 %[[#bb29:]] 0 %[[#bb30:]] 1 %[[#bb31:]] 2 %[[#bb32:]]
48; CHECK:    %[[#bb28:]] = OpLabel
49; CHECK:                  OpBranch %[[#bb27:]]
50; CHECK:    %[[#bb29:]] = OpLabel
51; CHECK:                  OpBranch %[[#bb27:]]
52; CHECK:    %[[#bb30:]] = OpLabel
53; CHECK:                  OpBranch %[[#bb27:]]
54; CHECK:    %[[#bb31:]] = OpLabel
55; CHECK:                  OpBranch %[[#bb27:]]
56; CHECK:    %[[#bb32:]] = OpLabel
57; CHECK:                  OpBranch %[[#bb27:]]
58; CHECK:    %[[#bb27:]] = OpLabel
59; CHECK:                  OpSelectionMerge %[[#bb33:]] None
60; CHECK:                  OpSwitch %[[#]] %[[#bb33:]] 4294967292 %[[#bb34:]] 4 %[[#bb35:]]
61; CHECK:    %[[#bb34:]] = OpLabel
62; CHECK:                  OpBranch %[[#bb33:]]
63; CHECK:    %[[#bb35:]] = OpLabel
64; CHECK:                  OpBranch %[[#bb33:]]
65; CHECK:    %[[#bb33:]] = OpLabel
66; CHECK:                  OpReturn
67; CHECK:                  OpFunctionEnd
68; CHECK: %[[#func_23:]] = OpFunction %[[#void:]] None %[[#]]
69; CHECK:    %[[#bb36:]] = OpLabel
70; CHECK:                  OpReturn
71; CHECK:                  OpFunctionEnd
72
73
74
75target datalayout = "e-i64:64-v16:16-v24:32-v32:32-v48:64-v96:128-v192:256-v256:256-v512:512-v1024:1024-G1"
76target triple = "spirv-unknown-vulkan1.3-compute"
77
78; Function Attrs: convergent noinline norecurse nounwind optnone
79define spir_func noundef i32 @_Z3foov() #0 {
80entry:
81  %0 = call token @llvm.experimental.convergence.entry()
82  ret i32 200
83}
84
85; Function Attrs: convergent nocallback nofree nosync nounwind willreturn memory(none)
86declare token @llvm.experimental.convergence.entry() #1
87
88; Function Attrs: convergent noinline norecurse nounwind optnone
89define internal spir_func void @main() #0 {
90entry:
91  %0 = call token @llvm.experimental.convergence.entry()
92  %result = alloca i32, align 4
93  %a = alloca i32, align 4
94  %c = alloca i32, align 4
95  store i32 0, ptr %a, align 4
96  %1 = load i32, ptr %a, align 4
97  switch i32 %1, label %sw.default [
98    i32 -3, label %sw.bb
99    i32 0, label %sw.bb1
100    i32 1, label %sw.bb2
101    i32 2, label %sw.bb3
102  ]
103
104sw.bb:                                            ; preds = %entry
105  store i32 -300, ptr %result, align 4
106  br label %sw.epilog
107
108sw.bb1:                                           ; preds = %entry
109  store i32 0, ptr %result, align 4
110  br label %sw.epilog
111
112sw.bb2:                                           ; preds = %entry
113  store i32 100, ptr %result, align 4
114  br label %sw.epilog
115
116sw.bb3:                                           ; preds = %entry
117  %call4 = call spir_func noundef i32 @_Z3foov() #3 [ "convergencectrl"(token %0) ]
118  store i32 %call4, ptr %result, align 4
119  br label %sw.epilog
120
121sw.default:                                       ; preds = %entry
122  store i32 777, ptr %result, align 4
123  br label %sw.epilog
124
125sw.epilog:                                        ; preds = %sw.default, %sw.bb3, %sw.bb2, %sw.bb1, %sw.bb
126  %2 = load i32, ptr %a, align 4
127  store i32 %2, ptr %c, align 4
128  %3 = load i32, ptr %c, align 4
129  switch i32 %3, label %sw.epilog7 [
130    i32 -4, label %sw.bb5
131    i32 4, label %sw.bb6
132  ]
133
134sw.bb5:                                           ; preds = %sw.epilog
135  store i32 -400, ptr %result, align 4
136  br label %sw.epilog7
137
138sw.bb6:                                           ; preds = %sw.epilog
139  store i32 400, ptr %result, align 4
140  br label %sw.epilog7
141
142sw.epilog7:                                       ; preds = %sw.epilog, %sw.bb6, %sw.bb5
143  ret void
144}
145
146; Function Attrs: convergent norecurse
147define void @main.1() #2 {
148entry:
149  call void @main()
150  ret void
151}
152
153attributes #0 = { convergent noinline norecurse nounwind optnone "frame-pointer"="all" "no-trapping-math"="true" "stack-protector-buffer-size"="8" }
154attributes #1 = { convergent nocallback nofree nosync nounwind willreturn memory(none) }
155attributes #2 = { convergent norecurse "frame-pointer"="all" "hlsl.numthreads"="1,1,1" "hlsl.shader"="compute" "no-trapping-math"="true" "stack-protector-buffer-size"="8" }
156attributes #3 = { convergent }
157
158!llvm.module.flags = !{!0, !1, !2}
159
160
161!0 = !{i32 1, !"wchar_size", i32 4}
162!1 = !{i32 4, !"dx.disable_optimizations", i32 1}
163!2 = !{i32 7, !"frame-pointer", i32 2}
164
165
166