llvm/test/CodeGen/SPIRV/structurizer/return-early.ll

   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 ; CHECK-DAG:         OpName %[[#reg_0:]] "reg2"
   5 ; CHECK-DAG:         OpName %[[#reg_1:]] "reg1"
   6
   7 target datalayout = "e-i64:64-v16:16-v24:32-v32:32-v48:64-v96:128-v192:256-v256:256-v512:512-v1024:1024-G1"
   8 target triple = "spirv-unknown-vulkan1.3-compute"
   9
  10 define internal spir_func void @main() #0 {
  11 ; CHECK: %[[#entry:]] = OpLabel
  12 ; CHECK:                OpBranch %[[#while_cond:]]
  13 entry:
  14   %0 = call token @llvm.experimental.convergence.entry()
  15   %a = alloca i32, align 4
  16   %b = alloca i32, align 4
  17   %cond = alloca i32, align 4
  18   %i = alloca i32, align 4
  19   store i32 1, ptr %cond, align 4
  20   br label %while.cond
  21
  22 ; CHECK: %[[#while_cond]] = OpLabel
  23 ; CHECK:                    OpSelectionMerge %[[#while_end:]] None
  24 ; CHECK:                    OpBranchConditional %[[#cond:]] %[[#while_body:]] %[[#while_end]]
  25 while.cond:
  26   %1 = call token @llvm.experimental.convergence.loop() [ "convergencectrl"(token %0) ]
  27   %2 = load i32, ptr %cond, align 4
  28   %tobool = icmp ne i32 %2, 0
  29   br i1 %tobool, label %while.body, label %while.end
  30
  31 ; CHECK: %[[#while_body]] = OpLabel
  32 ; CHECK:                    OpSelectionMerge %[[#switch_end:]] None
  33 ; CHECK:                    OpSwitch %[[#cond:]] %[[#switch_end]] 1 %[[#case_1:]] 2 %[[#case_2:]] 5 %[[#case_5:]]
  34 while.body:
  35   %3 = load i32, ptr %b, align 4
  36   switch i32 %3, label %sw.default [
  37     i32 1, label %sw.bb
  38     i32 2, label %sw.bb1
  39     i32 5, label %sw.bb2
  40   ]
  41
  42 ; CHECK: %[[#case_5]] = OpLabel
  43 ; CHECK:                OpStore %[[#reg_0]] %[[#]]
  44 ; CHECK:                OpBranch %[[#switch_end]]
  45 sw.bb2:
  46   store i32 5, ptr %a, align 4
  47   br label %while.end
  48
  49 ; CHECK: %[[#case_2]] = OpLabel
  50 ; CHECK:                OpStore %[[#reg_0]] %[[#]]
  51 ; CHECK:                OpBranch %[[#switch_end]]
  52 sw.bb1:
  53   store i32 3, ptr %a, align 4
  54   br label %while.end
  55
  56 ; CHECK: %[[#case_1]] = OpLabel
  57 ; CHECK:                OpStore %[[#reg_0]] %[[#]]
  58 ; CHECK:                OpBranch %[[#switch_end]]
  59 sw.bb:
  60   store i32 1, ptr %a, align 4
  61   br label %while.end
  62
  63 ; CHECK: %[[#switch_end]] = OpLabel
  64 ; CHECK:       %[[#val:]] = OpLoad %[[#]] %[[#reg_0]]
  65 ; CHECK:       %[[#tmp:]] = OpIEqual %[[#type:]] %[[#]] %[[#val]]
  66 ; CHECK:                    OpBranchConditional %[[#tmp]] %[[#sw_default:]] %[[#while_end]]
  67
  68 ; CHECK: %[[#sw_default]] = OpLabel
  69 ; CHECK:                    OpStore %[[#]] %[[#B:]] Aligned 4
  70 ; CHECK:                    OpBranch %[[#for_cond:]]
  71 sw.default:
  72   store i32 0, ptr %i, align 4
  73   br label %for.cond
  74
  75 ; CHECK: %[[#for_cond]] = OpLabel
  76 ; CHECK:                  OpStore %[[#reg_1]] %[[#]]
  77 ; CHECK:                  OpSelectionMerge %[[#for_merge:]] None
  78 ; CHECK-NEXT:             OpBranchConditional %[[#cond:]] %[[#for_merge]] %[[#for_end:]]
  79 for.cond:
  80   %4 = call token @llvm.experimental.convergence.loop() [ "convergencectrl"(token %1) ]
  81   %5 = load i32, ptr %i, align 4
  82   %cmp = icmp slt i32 %5, 10
  83   br i1 %cmp, label %for.body, label %for.end
  84
  85 ; CHECK: %[[#for_end]] = OpLabel
  86 ; CHECK:                 OpStore %[[#reg_1]] %[[#]]
  87 ; CHECK:                 OpBranch %[[#for_merge]]
  88 for.end:
  89   br label %while.end
  90
  91 ; CHECK: %[[#for_merge]] = OpLabel
  92 ; CHECK:      %[[#val:]] = OpLoad %[[#]] %[[#reg_1]]
  93 ; CHECK:      %[[#tmp:]] = OpIEqual %[[#type:]] %[[#]] %[[#val]]
  94 ; CHECK:                   OpBranchConditional %[[#tmp]] %[[#for_body:]] %[[#while_end]]
  95
  96 ; CHECK: %[[#for_body]] = OpLabel
  97 ; CHECK:                  OpSelectionMerge %[[#if_merge:]] None
  98 ; CHECK:                  OpBranchConditional %[[#cond:]] %[[#if_merge]] %[[#if_else:]]
  99 for.body:
 100   %6 = load i32, ptr %cond, align 4
 101   %tobool3 = icmp ne i32 %6, 0
 102   br i1 %tobool3, label %if.then, label %if.else
 103
 104 ; CHECK: %[[#if_else]] = OpLabel
 105 ; CHECK:                 OpBranch %[[#if_merge]]
 106 if.else:
 107   br label %while.end
 108
 109 ; CHECK: %[[#if_merge]] = OpLabel
 110 ; CHECK:                  OpBranch %[[#while_end]]
 111 if.then:
 112   br label %while.end
 113
 114 ; CHECK: %[[#while_end]] = OpLabel
 115 ; CHECK:                   OpReturn
 116 while.end:
 117   ret void
 118
 119 ; CHECK-NOT: %[[#for_inc:]] = OpLabel
 120 ; This block is not emitted since it's unreachable.
 121 for.inc:
 122   %7 = load i32, ptr %i, align 4
 123   %inc = add nsw i32 %7, 1
 124   store i32 %inc, ptr %i, align 4
 125   br label %for.cond
 126
 127 }
 128
 129 declare token @llvm.experimental.convergence.entry() #1
 130 declare token @llvm.experimental.convergence.loop() #1
 131
 132 attributes #0 = { convergent noinline norecurse nounwind optnone "frame-pointer"="all" "no-trapping-math"="true" "stack-protector-buffer-size"="8" }
 133 attributes #1 = { convergent nocallback nofree nosync nounwind willreturn memory(none) }
 134
 135 !llvm.module.flags = !{!0, !1, !2}
 136
 137 !0 = !{i32 1, !"wchar_size", i32 4}
 138 !1 = !{i32 4, !"dx.disable_optimizations", i32 1}
 139 !2 = !{i32 7, !"frame-pointer", i32 2}