llvm/test/Transforms/OpenMP/is_spmd_exec_mode_fold.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --function-signature --check-globals
   2 ; RUN: opt -S -passes=openmp-opt < %s | FileCheck %s
   3 target triple = "nvptx64"
   4
   5 %struct.ident_t = type { i32, i32, i32, i32, i8* }
   6
   7 @is_spmd_exec_mode = weak constant i8 0
   8 @will_be_spmd_exec_mode = weak constant i8 1
   9 @non_spmd_exec_mode = weak constant i8 1
  10 @will_not_be_spmd_exec_mode = weak constant i8 1
  11 @G = external global i8
  12 @llvm.compiler.used = appending global [4 x i8*] [i8* @is_spmd_exec_mode, i8* @will_be_spmd_exec_mode, i8* @non_spmd_exec_mode, i8* @will_not_be_spmd_exec_mode ], section "llvm.metadata"
  13
  14 ;.
  15 ; CHECK: @[[IS_SPMD_EXEC_MODE:[a-zA-Z0-9_$"\\.-]+]] = weak constant i8 0
  16 ; CHECK: @[[WILL_BE_SPMD_EXEC_MODE:[a-zA-Z0-9_$"\\.-]+]] = weak constant i8 3
  17 ; CHECK: @[[NON_SPMD_EXEC_MODE:[a-zA-Z0-9_$"\\.-]+]] = weak constant i8 1
  18 ; CHECK: @[[WILL_NOT_BE_SPMD_EXEC_MODE:[a-zA-Z0-9_$"\\.-]+]] = weak constant i8 1
  19 ; CHECK: @[[G:[a-zA-Z0-9_$"\\.-]+]] = external global i8
  20 ; CHECK: @[[LLVM_COMPILER_USED:[a-zA-Z0-9_$"\\.-]+]] = appending global [4 x i8*] [i8* @is_spmd_exec_mode, i8* @will_be_spmd_exec_mode, i8* @non_spmd_exec_mode, i8* @will_not_be_spmd_exec_mode], section "llvm.metadata"
  21 ;.
  22 define weak void @is_spmd() {
  23 ; CHECK-LABEL: define {{[^@]+}}@is_spmd() {
  24 ; CHECK-NEXT:    [[I:%.*]] = call i32 @__kmpc_target_init(%struct.ident_t* null, i8 2, i1 false, i1 false)
  25 ; CHECK-NEXT:    call void @is_spmd_helper1()
  26 ; CHECK-NEXT:    call void @is_spmd_helper2()
  27 ; CHECK-NEXT:    call void @is_mixed_helper()
  28 ; CHECK-NEXT:    call void @__kmpc_target_deinit(%struct.ident_t* null, i8 2, i1 false)
  29 ; CHECK-NEXT:    ret void
  30 ;
  31   %i = call i32 @__kmpc_target_init(%struct.ident_t* null, i8 2, i1 false, i1 false)
  32   call void @is_spmd_helper1()
  33   call void @is_spmd_helper2()
  34   call void @is_mixed_helper()
  35   call void @__kmpc_target_deinit(%struct.ident_t* null, i8 2, i1 false)
  36   ret void
  37 }
  38
  39 define weak void @will_be_spmd() {
  40 ; CHECK-LABEL: define {{[^@]+}}@will_be_spmd() {
  41 ; CHECK-NEXT:  entry:
  42 ; CHECK-NEXT:    [[CAPTURED_VARS_ADDRS:%.*]] = alloca [0 x i8*], align 8
  43 ; CHECK-NEXT:    [[I:%.*]] = call i32 @__kmpc_target_init(%struct.ident_t* null, i8 2, i1 false, i1 false)
  44 ; CHECK-NEXT:    [[EXEC_USER_CODE:%.*]] = icmp eq i32 [[I]], -1
  45 ; CHECK-NEXT:    br i1 [[EXEC_USER_CODE]], label [[USER_CODE_ENTRY:%.*]], label [[COMMON_RET:%.*]]
  46 ; CHECK:       common.ret:
  47 ; CHECK-NEXT:    ret void
  48 ; CHECK:       user_code.entry:
  49 ; CHECK-NEXT:    [[TMP0:%.*]] = call i32 @__kmpc_global_thread_num(%struct.ident_t* null) #[[ATTR2:[0-9]+]]
  50 ; CHECK-NEXT:    [[TMP1:%.*]] = bitcast [0 x i8*]* [[CAPTURED_VARS_ADDRS]] to i8**
  51 ; CHECK-NEXT:    call void @is_spmd_helper2()
  52 ; CHECK-NEXT:    call void @__kmpc_parallel_51(%struct.ident_t* null, i32 [[TMP0]], i32 1, i32 -1, i32 -1, i8* bitcast (void (i32*, i32*)* @__omp_outlined__ to i8*), i8* bitcast (void (i16, i32)* @__omp_outlined___wrapper to i8*), i8** [[TMP1]], i64 0)
  53 ; CHECK-NEXT:    call void @__kmpc_target_deinit(%struct.ident_t* null, i8 2, i1 false)
  54 ; CHECK-NEXT:    ret void
  55 ;
  56 entry:
  57   %captured_vars_addrs = alloca [0 x i8*], align 8
  58   %i = call i32 @__kmpc_target_init(%struct.ident_t* null, i8 1, i1 true, i1 true)
  59   %exec_user_code = icmp eq i32 %i, -1
  60   br i1 %exec_user_code, label %user_code.entry, label %common.ret
  61
  62 common.ret:
  63   ret void
  64
  65 user_code.entry:
  66   %0 = call i32 @__kmpc_global_thread_num(%struct.ident_t* null)
  67   %1 = bitcast [0 x i8*]* %captured_vars_addrs to i8**
  68   call void @is_spmd_helper2()
  69   call void @__kmpc_parallel_51(%struct.ident_t* null, i32 %0, i32 1, i32 -1, i32 -1, i8* bitcast (void (i32*, i32*)* @__omp_outlined__ to i8*), i8* bitcast (void (i16, i32)* @__omp_outlined___wrapper to i8*), i8** %1, i64 0)
  70   call void @__kmpc_target_deinit(%struct.ident_t* null, i8 1, i1 true)
  71   ret void
  72 }
  73
  74 define weak void @non_spmd() {
  75 ; CHECK-LABEL: define {{[^@]+}}@non_spmd() {
  76 ; CHECK-NEXT:    [[I:%.*]] = call i32 @__kmpc_target_init(%struct.ident_t* null, i8 1, i1 false, i1 false)
  77 ; CHECK-NEXT:    call void @is_generic_helper1()
  78 ; CHECK-NEXT:    call void @is_generic_helper2()
  79 ; CHECK-NEXT:    call void @is_mixed_helper()
  80 ; CHECK-NEXT:    call void @__kmpc_target_deinit(%struct.ident_t* null, i8 1, i1 false)
  81 ; CHECK-NEXT:    ret void
  82 ;
  83   %i = call i32 @__kmpc_target_init(%struct.ident_t* null, i8 1, i1 false, i1 false)
  84   call void @is_generic_helper1()
  85   call void @is_generic_helper2()
  86   call void @is_mixed_helper()
  87   call void @__kmpc_target_deinit(%struct.ident_t* null, i8 1, i1 false)
  88   ret void
  89 }
  90
  91 define weak void @will_not_be_spmd() {
  92 ; CHECK-LABEL: define {{[^@]+}}@will_not_be_spmd() {
  93 ; CHECK-NEXT:    [[I:%.*]] = call i32 @__kmpc_target_init(%struct.ident_t* null, i8 1, i1 false, i1 false)
  94 ; CHECK-NEXT:    call void @is_generic_helper1()
  95 ; CHECK-NEXT:    call void @is_generic_helper2()
  96 ; CHECK-NEXT:    call void @is_mixed_helper()
  97 ; CHECK-NEXT:    call void @__kmpc_target_deinit(%struct.ident_t* null, i8 1, i1 false)
  98 ; CHECK-NEXT:    ret void
  99 ;
 100   %i = call i32 @__kmpc_target_init(%struct.ident_t* null, i8 1, i1 false, i1 false)
 101   call void @is_generic_helper1()
 102   call void @is_generic_helper2()
 103   call void @is_mixed_helper()
 104   call void @__kmpc_target_deinit(%struct.ident_t* null, i8 1, i1 false)
 105   ret void
 106 }
 107
 108 define internal void @is_spmd_helper1() {
 109 ; CHECK-LABEL: define {{[^@]+}}@is_spmd_helper1() {
 110 ; CHECK-NEXT:    store i8 1, i8* @G, align 1
 111 ; CHECK-NEXT:    ret void
 112 ;
 113   %isSPMD = call i8 @__kmpc_is_spmd_exec_mode()
 114   store i8 %isSPMD, i8* @G
 115   ret void
 116 }
 117
 118 define internal void @is_spmd_helper2() {
 119 ; CHECK-LABEL: define {{[^@]+}}@is_spmd_helper2() {
 120 ; CHECK-NEXT:    br label [[F:%.*]]
 121 ; CHECK:       t:
 122 ; CHECK-NEXT:    unreachable
 123 ; CHECK:       f:
 124 ; CHECK-NEXT:    ret void
 125 ;
 126   %isSPMD = call i8 @__kmpc_is_spmd_exec_mode()
 127   %c = icmp eq i8 %isSPMD, 0
 128   br i1 %c, label %t, label %f
 129 t:
 130   call void @spmd_compatible()
 131   ret void
 132 f:
 133   ret void
 134 }
 135
 136 define internal void @is_generic_helper1() {
 137 ; CHECK-LABEL: define {{[^@]+}}@is_generic_helper1() {
 138 ; CHECK-NEXT:    store i8 0, i8* @G, align 1
 139 ; CHECK-NEXT:    ret void
 140 ;
 141   %isSPMD = call i8 @__kmpc_is_spmd_exec_mode()
 142   store i8 %isSPMD, i8* @G
 143   ret void
 144 }
 145
 146 define internal void @is_generic_helper2() {
 147 ; CHECK-LABEL: define {{[^@]+}}@is_generic_helper2() {
 148 ; CHECK-NEXT:    br label [[T:%.*]]
 149 ; CHECK:       t:
 150 ; CHECK-NEXT:    call void @foo()
 151 ; CHECK-NEXT:    ret void
 152 ; CHECK:       f:
 153 ; CHECK-NEXT:    call void @bar()
 154 ; CHECK-NEXT:    ret void
 155 ;
 156   %isSPMD = call i8 @__kmpc_is_spmd_exec_mode()
 157   %c = icmp eq i8 %isSPMD, 0
 158   br i1 %c, label %t, label %f
 159 t:
 160   call void @foo()
 161   ret void
 162 f:
 163   call void @bar()
 164   ret void
 165 }
 166
 167 define internal void @is_mixed_helper() {
 168 ; CHECK-LABEL: define {{[^@]+}}@is_mixed_helper() {
 169 ; CHECK-NEXT:    [[ISSPMD:%.*]] = call i8 @__kmpc_is_spmd_exec_mode()
 170 ; CHECK-NEXT:    store i8 [[ISSPMD]], i8* @G, align 1
 171 ; CHECK-NEXT:    ret void
 172 ;
 173   %isSPMD = call i8 @__kmpc_is_spmd_exec_mode()
 174   store i8 %isSPMD, i8* @G
 175   ret void
 176 }
 177
 178 define internal void @__omp_outlined__(i32* noalias %.global_tid., i32* noalias %.bound_tid.) {
 179 ; CHECK-LABEL: define {{[^@]+}}@__omp_outlined__
 180 ; CHECK-SAME: (i32* noalias [[DOTGLOBAL_TID_:%.*]], i32* noalias [[DOTBOUND_TID_:%.*]]) {
 181 ; CHECK-NEXT:  entry:
 182 ; CHECK-NEXT:    ret void
 183 ;
 184 entry:
 185   ret void
 186 }
 187
 188 define internal void @__omp_outlined___wrapper(i16 zeroext %0, i32 %1) {
 189 ; CHECK-LABEL: define {{[^@]+}}@__omp_outlined___wrapper
 190 ; CHECK-SAME: (i16 zeroext [[TMP0:%.*]], i32 [[TMP1:%.*]]) {
 191 ; CHECK-NEXT:  entry:
 192 ; CHECK-NEXT:    ret void
 193 ;
 194 entry:
 195   ret void
 196 }
 197
 198 declare void @spmd_compatible() "llvm.assume"="ompx_spmd_amenable"
 199 declare i8 @__kmpc_is_spmd_exec_mode()
 200 declare i32 @__kmpc_target_init(%struct.ident_t*, i8, i1 zeroext, i1 zeroext)
 201 declare void @__kmpc_target_deinit(%struct.ident_t* nocapture readnone, i8, i1 zeroext)
 202 declare void @__kmpc_parallel_51(%struct.ident_t*, i32, i32, i32, i32, i8*, i8*, i8**, i64)
 203 declare i32 @__kmpc_global_thread_num(%struct.ident_t*)
 204 declare void @foo()
 205 declare void @bar()
 206
 207 !llvm.module.flags = !{!0, !1}
 208 !nvvm.annotations = !{!2, !3, !4, !5}
 209
 210 !0 = !{i32 7, !"openmp", i32 50}
 211 !1 = !{i32 7, !"openmp-device", i32 50}
 212 !2 = !{void ()* @is_spmd, !"kernel", i32 1}
 213 !3 = !{void ()* @will_be_spmd, !"kernel", i32 1}
 214 !4 = !{void ()* @non_spmd, !"kernel", i32 1}
 215 !5 = !{void ()* @will_not_be_spmd, !"kernel", i32 1}
 216 ;.
 217 ; CHECK: attributes #[[ATTR0:[0-9]+]] = { "llvm.assume"="ompx_spmd_amenable" }
 218 ; CHECK: attributes #[[ATTR1:[0-9]+]] = { alwaysinline }
 219 ; CHECK: attributes #[[ATTR2]] = { nounwind }
 220 ;.
 221 ; CHECK: [[META0:![0-9]+]] = !{i32 7, !"openmp", i32 50}
 222 ; CHECK: [[META1:![0-9]+]] = !{i32 7, !"openmp-device", i32 50}
 223 ; CHECK: [[META2:![0-9]+]] = !{void ()* @is_spmd, !"kernel", i32 1}
 224 ; CHECK: [[META3:![0-9]+]] = !{void ()* @will_be_spmd, !"kernel", i32 1}
 225 ; CHECK: [[META4:![0-9]+]] = !{void ()* @non_spmd, !"kernel", i32 1}
 226 ; CHECK: [[META5:![0-9]+]] = !{void ()* @will_not_be_spmd, !"kernel", i32 1}
 227 ;.