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 3target 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;. 22define 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 39define 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; 56entry: 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 62common.ret: 63 ret void 64 65user_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 74define 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 91define 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 108define 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 118define 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 129t: 130 call void @spmd_compatible() 131 ret void 132f: 133 ret void 134} 135 136define 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 146define 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 159t: 160 call void @foo() 161 ret void 162f: 163 call void @bar() 164 ret void 165} 166 167define 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 178define 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; 184entry: 185 ret void 186} 187 188define 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; 194entry: 195 ret void 196} 197 198declare void @spmd_compatible() "llvm.assume"="ompx_spmd_amenable" 199declare i8 @__kmpc_is_spmd_exec_mode() 200declare i32 @__kmpc_target_init(%struct.ident_t*, i8, i1 zeroext, i1 zeroext) 201declare void @__kmpc_target_deinit(%struct.ident_t* nocapture readnone, i8, i1 zeroext) 202declare void @__kmpc_parallel_51(%struct.ident_t*, i32, i32, i32, i32, i8*, i8*, i8**, i64) 203declare i32 @__kmpc_global_thread_num(%struct.ident_t*) 204declare void @foo() 205declare 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;. 228