clang/test/OpenMP/master_taskloop_codegen.cpp

   1 // RUN: %clang_cc1 -verify -triple x86_64-apple-darwin10 -fopenmp -x c++ -emit-llvm %s -o - | FileCheck %s
   2 // RUN: %clang_cc1 -fopenmp -x c++ -triple x86_64-apple-darwin10 -emit-pch -o %t %s
   3 // RUN: %clang_cc1 -fopenmp -x c++ -triple x86_64-apple-darwin10 -include-pch %t -verify %s -emit-llvm -o - | FileCheck %s
   4
   5 // RUN: %clang_cc1 -verify -triple x86_64-apple-darwin10 -fopenmp-simd -x c++ -emit-llvm %s -o - | FileCheck --check-prefix SIMD-ONLY0 %s
   6 // RUN: %clang_cc1 -fopenmp-simd -x c++ -triple x86_64-apple-darwin10 -emit-pch -o %t %s
   7 // RUN: %clang_cc1 -fopenmp-simd -x c++ -triple x86_64-apple-darwin10 -include-pch %t -verify %s -emit-llvm -o - | FileCheck --check-prefix SIMD-ONLY0 %s
   8 // SIMD-ONLY0-NOT: {{__kmpc|__tgt}}
   9 // expected-no-diagnostics
  10 #ifndef HEADER
  11 #define HEADER
  12
  13 // CHECK-LABEL: @main
  14 int main(int argc, char **argv) {
  15 // CHECK: [[GTID:%.+]] = call i32 @__kmpc_global_thread_num(ptr [[DEFLOC:@.+]])
  16 // CHECK: call ptr @__kmpc_omp_task_alloc(ptr [[DEFLOC]], i32 [[GTID]],
  17 // CHECK: call i32 @__kmpc_omp_task(ptr [[DEFLOC]], i32 [[GTID]],
  18 #pragma omp task
  19   ;
  20 // CHECK:       [[RES:%.+]] = call {{.*}}i32 @__kmpc_master(ptr [[DEFLOC]], i32 [[GTID]])
  21 // CHECK-NEXT:  [[IS_MASTER:%.+]] = icmp ne i32 [[RES]], 0
  22 // CHECK-NEXT:  br i1 [[IS_MASTER]], label {{%?}}[[THEN:.+]], label {{%?}}[[EXIT:.+]]
  23 // CHECK:       [[THEN]]
  24 // CHECK: call void @__kmpc_taskgroup(ptr [[DEFLOC]], i32 [[GTID]])
  25 // CHECK: [[TASKV:%.+]] = call ptr @__kmpc_omp_task_alloc(ptr [[DEFLOC]], i32 [[GTID]], i32 33, i64 80, i64 1, ptr [[TASK1:@.+]])
  26 // CHECK: [[TASK_DATA:%.+]] = getelementptr inbounds nuw %{{.+}}, ptr [[TASKV]], i32 0, i32 0
  27 // CHECK: [[DOWN:%.+]] = getelementptr inbounds nuw [[TD_TY:%.+]], ptr [[TASK_DATA]], i32 0, i32 5
  28 // CHECK: store i64 0, ptr [[DOWN]],
  29 // CHECK: [[UP:%.+]] = getelementptr inbounds nuw [[TD_TY]], ptr [[TASK_DATA]], i32 0, i32 6
  30 // CHECK: store i64 9, ptr [[UP]],
  31 // CHECK: [[ST:%.+]] = getelementptr inbounds nuw [[TD_TY]], ptr [[TASK_DATA]], i32 0, i32 7
  32 // CHECK: store i64 1, ptr [[ST]],
  33 // CHECK: [[ST_VAL:%.+]] = load i64, ptr [[ST]],
  34 // CHECK: call void @__kmpc_taskloop(ptr [[DEFLOC]], i32 [[GTID]], ptr [[TASKV]], i32 1, ptr [[DOWN]], ptr [[UP]], i64 [[ST_VAL]], i32 1, i32 0, i64 0, ptr null)
  35 // CHECK: call void @__kmpc_end_taskgroup(ptr [[DEFLOC]], i32 [[GTID]])
  36 // CHECK-NEXT:  call {{.*}}void @__kmpc_end_master(ptr [[DEFLOC]], i32 [[GTID]])
  37 // CHECK-NEXT:  br label {{%?}}[[EXIT]]
  38 // CHECK:       [[EXIT]]
  39 #pragma omp master taskloop priority(argc)
  40   for (int i = 0; i < 10; ++i)
  41     ;
  42 // CHECK:       [[RES:%.+]] = call {{.*}}i32 @__kmpc_master(ptr [[DEFLOC]], i32 [[GTID]])
  43 // CHECK-NEXT:  [[IS_MASTER:%.+]] = icmp ne i32 [[RES]], 0
  44 // CHECK-NEXT:  br i1 [[IS_MASTER]], label {{%?}}[[THEN:.+]], label {{%?}}[[EXIT:.+]]
  45 // CHECK:       [[THEN]]
  46 // CHECK: [[TASKV:%.+]] = call ptr @__kmpc_omp_task_alloc(ptr [[DEFLOC]], i32 [[GTID]], i32 1, i64 80, i64 1, ptr [[TASK2:@.+]])
  47 // CHECK: [[TASK_DATA:%.+]] = getelementptr inbounds nuw %{{.+}}, ptr [[TASKV]], i32 0, i32 0
  48 // CHECK: [[DOWN:%.+]] = getelementptr inbounds nuw [[TD_TY:%.+]], ptr [[TASK_DATA]], i32 0, i32 5
  49 // CHECK: store i64 0, ptr [[DOWN]],
  50 // CHECK: [[UP:%.+]] = getelementptr inbounds nuw [[TD_TY]], ptr [[TASK_DATA]], i32 0, i32 6
  51 // CHECK: store i64 9, ptr [[UP]],
  52 // CHECK: [[ST:%.+]] = getelementptr inbounds nuw [[TD_TY]], ptr [[TASK_DATA]], i32 0, i32 7
  53 // CHECK: store i64 1, ptr [[ST]],
  54 // CHECK: [[ST_VAL:%.+]] = load i64, ptr [[ST]],
  55 // CHECK: [[GRAINSIZE:%.+]] = zext i32 %{{.+}} to i64
  56 // CHECK: call void @__kmpc_taskloop(ptr [[DEFLOC]], i32 [[GTID]], ptr [[TASKV]], i32 1, ptr [[DOWN]], ptr [[UP]], i64 [[ST_VAL]], i32 1, i32 1, i64 [[GRAINSIZE]], ptr null)
  57 // CHECK-NEXT:  call {{.*}}void @__kmpc_end_master(ptr [[DEFLOC]], i32 [[GTID]])
  58 // CHECK-NEXT:  br label {{%?}}[[EXIT]]
  59 // CHECK:       [[EXIT]]
  60 #pragma omp master taskloop nogroup grainsize(argc)
  61   for (int i = 0; i < 10; ++i)
  62     ;
  63 // CHECK:       [[RES:%.+]] = call {{.*}}i32 @__kmpc_master(ptr [[DEFLOC]], i32 [[GTID]])
  64 // CHECK-NEXT:  [[IS_MASTER:%.+]] = icmp ne i32 [[RES]], 0
  65 // CHECK-NEXT:  br i1 [[IS_MASTER]], label {{%?}}[[THEN:.+]], label {{%?}}[[EXIT:.+]]
  66 // CHECK:       [[THEN]]
  67 // CHECK: call void @__kmpc_taskgroup(ptr [[DEFLOC]], i32 [[GTID]])
  68 // CHECK: [[TASKV:%.+]] = call ptr @__kmpc_omp_task_alloc(ptr [[DEFLOC]], i32 [[GTID]], i32 1, i64 80, i64 16, ptr [[TASK3:@.+]])
  69 // CHECK: [[TASK_DATA:%.+]] = getelementptr inbounds nuw %{{.+}}, ptr [[TASKV]], i32 0, i32 0
  70 // CHECK: [[IF:%.+]] = icmp ne i32 %{{.+}}, 0
  71 // CHECK: [[IF_INT:%.+]] = sext i1 [[IF]] to i32
  72 // CHECK: [[DOWN:%.+]] = getelementptr inbounds nuw [[TD_TY:%.+]], ptr [[TASK_DATA]], i32 0, i32 5
  73 // CHECK: store i64 0, ptr [[DOWN]],
  74 // CHECK: [[UP:%.+]] = getelementptr inbounds nuw [[TD_TY]], ptr [[TASK_DATA]], i32 0, i32 6
  75 // CHECK: store i64 %{{.+}}, ptr [[UP]],
  76 // CHECK: [[ST:%.+]] = getelementptr inbounds nuw [[TD_TY]], ptr [[TASK_DATA]], i32 0, i32 7
  77 // CHECK: store i64 1, ptr [[ST]],
  78 // CHECK: [[ST_VAL:%.+]] = load i64, ptr [[ST]],
  79 // CHECK: call void @__kmpc_taskloop(ptr [[DEFLOC]], i32 [[GTID]], ptr [[TASKV]], i32 [[IF_INT]], ptr [[DOWN]], ptr [[UP]], i64 [[ST_VAL]], i32 1, i32 2, i64 4, ptr null)
  80 // CHECK: call void @__kmpc_end_taskgroup(ptr [[DEFLOC]], i32 [[GTID]])
  81 // CHECK-NEXT:  call {{.*}}void @__kmpc_end_master(ptr [[DEFLOC]], i32 [[GTID]])
  82 // CHECK-NEXT:  br label {{%?}}[[EXIT]]
  83 // CHECK:       [[EXIT]]
  84   int i;
  85 #pragma omp master taskloop if(argc) shared(argc, argv) collapse(2) num_tasks(4)
  86   for (i = 0; i < argc; ++i)
  87   for (int j = argc; j < argv[argc][argc]; ++j)
  88     ;
  89 // CHECK:       [[RES:%.+]] = call {{.*}}i32 @__kmpc_master(ptr [[DEFLOC]], i32 [[GTID]])
  90 // CHECK-NEXT:  [[IS_MASTER:%.+]] = icmp ne i32 [[RES]], 0
  91 // CHECK-NEXT:  br i1 [[IS_MASTER]], label {{%?}}[[THEN:.+]], label {{%?}}[[EXIT:.+]]
  92 // CHECK:       [[THEN]]
  93 // CHECK: call void @__kmpc_taskgroup(
  94 // CHECK: call ptr @__kmpc_omp_task_alloc(ptr @{{.+}}, i32 %{{.+}}, i32 1, i64 80, i64 1, ptr [[TASK_CANCEL:@.+]])
  95 // CHECK: call void @__kmpc_taskloop(
  96 // CHECK: call void @__kmpc_end_taskgroup(
  97 // CHECK-NEXT:  call {{.*}}void @__kmpc_end_master(ptr [[DEFLOC]], i32 [[GTID]])
  98 // CHECK-NEXT:  br label {{%?}}[[EXIT]]
  99 // CHECK:       [[EXIT]]
 100 #pragma omp master taskloop
 101   for (int i = 0; i < 10; ++i) {
 102 #pragma omp cancel taskgroup
 103 #pragma omp cancellation point taskgroup
 104   }
 105 }
 106
 107 // CHECK: define internal noundef i32 [[TASK1]](
 108 // CHECK: [[DOWN:%.+]] = getelementptr inbounds nuw [[TD_TY:%.+]], ptr %{{.+}}, i32 0, i32 5
 109 // CHECK: [[DOWN_VAL:%.+]] = load i64, ptr [[DOWN]],
 110 // CHECK: [[UP:%.+]] = getelementptr inbounds nuw [[TD_TY]], ptr %{{.+}}, i32 0, i32 6
 111 // CHECK: [[UP_VAL:%.+]] = load i64, ptr [[UP]],
 112 // CHECK: [[ST:%.+]] = getelementptr inbounds nuw [[TD_TY]], ptr %{{.+}}, i32 0, i32 7
 113 // CHECK: [[ST_VAL:%.+]] = load i64, ptr [[ST]],
 114 // CHECK: [[LITER:%.+]] = getelementptr inbounds nuw [[TD_TY]], ptr %{{.+}}, i32 0, i32 8
 115 // CHECK: [[LITER_VAL:%.+]] = load i32, ptr [[LITER]],
 116 // CHECK: store i64 [[DOWN_VAL]], ptr [[LB:%[^,]+]],
 117 // CHECK: store i64 [[UP_VAL]], ptr [[UB:%[^,]+]],
 118 // CHECK: store i64 [[ST_VAL]], ptr [[ST:%[^,]+]],
 119 // CHECK: store i32 [[LITER_VAL]], ptr [[LITER:%[^,]+]],
 120 // CHECK: [[LB_VAL:%.+]] = load i64, ptr [[LB]],
 121 // CHECK: [[LB_I32:%.+]] = trunc i64 [[LB_VAL]] to i32
 122 // CHECK: store i32 [[LB_I32]], ptr [[CNT:%.+]],
 123 // CHECK: br label
 124 // CHECK: [[VAL:%.+]] = load i32, ptr [[CNT]],
 125 // CHECK: [[VAL_I64:%.+]] = sext i32 [[VAL]] to i64
 126 // CHECK: [[UB_VAL:%.+]] = load i64, ptr [[UB]],
 127 // CHECK: [[CMP:%.+]] = icmp ule i64 [[VAL_I64]], [[UB_VAL]]
 128 // CHECK: br i1 [[CMP]], label %{{.+}}, label %{{.+}}
 129 // CHECK: load i32, ptr %
 130 // CHECK: store i32 %
 131 // CHECK: load i32, ptr %
 132 // CHECK: add nsw i32 %{{.+}}, 1
 133 // CHECK: store i32 %{{.+}}, ptr %
 134 // CHECK: br label %
 135 // CHECK: ret i32 0
 136
 137 // CHECK: define internal noundef i32 [[TASK2]](
 138 // CHECK: [[DOWN:%.+]] = getelementptr inbounds nuw [[TD_TY:%.+]], ptr %{{.+}}, i32 0, i32 5
 139 // CHECK: [[DOWN_VAL:%.+]] = load i64, ptr [[DOWN]],
 140 // CHECK: [[UP:%.+]] = getelementptr inbounds nuw [[TD_TY]], ptr %{{.+}}, i32 0, i32 6
 141 // CHECK: [[UP_VAL:%.+]] = load i64, ptr [[UP]],
 142 // CHECK: [[ST:%.+]] = getelementptr inbounds nuw [[TD_TY]], ptr %{{.+}}, i32 0, i32 7
 143 // CHECK: [[ST_VAL:%.+]] = load i64, ptr [[ST]],
 144 // CHECK: [[LITER:%.+]] = getelementptr inbounds nuw [[TD_TY]], ptr %{{.+}}, i32 0, i32 8
 145 // CHECK: [[LITER_VAL:%.+]] = load i32, ptr [[LITER]],
 146 // CHECK: store i64 [[DOWN_VAL]], ptr [[LB:%[^,]+]],
 147 // CHECK: store i64 [[UP_VAL]], ptr [[UB:%[^,]+]],
 148 // CHECK: store i64 [[ST_VAL]], ptr [[ST:%[^,]+]],
 149 // CHECK: store i32 [[LITER_VAL]], ptr [[LITER:%[^,]+]],
 150 // CHECK: [[LB_VAL:%.+]] = load i64, ptr [[LB]],
 151 // CHECK: [[LB_I32:%.+]] = trunc i64 [[LB_VAL]] to i32
 152 // CHECK: store i32 [[LB_I32]], ptr [[CNT:%.+]],
 153 // CHECK: br label
 154 // CHECK: [[VAL:%.+]] = load i32, ptr [[CNT]],
 155 // CHECK: [[VAL_I64:%.+]] = sext i32 [[VAL]] to i64
 156 // CHECK: [[UB_VAL:%.+]] = load i64, ptr [[UB]],
 157 // CHECK: [[CMP:%.+]] = icmp ule i64 [[VAL_I64]], [[UB_VAL]]
 158 // CHECK: br i1 [[CMP]], label %{{.+}}, label %{{.+}}
 159 // CHECK: load i32, ptr %
 160 // CHECK: store i32 %
 161 // CHECK: load i32, ptr %
 162 // CHECK: add nsw i32 %{{.+}}, 1
 163 // CHECK: store i32 %{{.+}}, ptr %
 164 // CHECK: br label %
 165 // CHECK: ret i32 0
 166
 167 // CHECK: define internal noundef i32 [[TASK3]](
 168 // CHECK: [[DOWN:%.+]] = getelementptr inbounds nuw [[TD_TY:%.+]], ptr %{{.+}}, i32 0, i32 5
 169 // CHECK: [[DOWN_VAL:%.+]] = load i64, ptr [[DOWN]],
 170 // CHECK: [[UP:%.+]] = getelementptr inbounds nuw [[TD_TY]], ptr %{{.+}}, i32 0, i32 6
 171 // CHECK: [[UP_VAL:%.+]] = load i64, ptr [[UP]],
 172 // CHECK: [[ST:%.+]] = getelementptr inbounds nuw [[TD_TY]], ptr %{{.+}}, i32 0, i32 7
 173 // CHECK: [[ST_VAL:%.+]] = load i64, ptr [[ST]],
 174 // CHECK: [[LITER:%.+]] = getelementptr inbounds nuw [[TD_TY]], ptr %{{.+}}, i32 0, i32 8
 175 // CHECK: [[LITER_VAL:%.+]] = load i32, ptr [[LITER]],
 176 // CHECK: store i64 [[DOWN_VAL]], ptr [[LB:%[^,]+]],
 177 // CHECK: store i64 [[UP_VAL]], ptr [[UB:%[^,]+]],
 178 // CHECK: store i64 [[ST_VAL]], ptr [[ST:%[^,]+]],
 179 // CHECK: store i32 [[LITER_VAL]], ptr [[LITER:%[^,]+]],
 180 // CHECK: [[LB_VAL:%.+]] = load i64, ptr [[LB]],
 181 // CHECK: store i64 [[LB_VAL]], ptr [[CNT:%.+]],
 182 // CHECK: br label
 183 // CHECK: ret i32 0
 184
 185 // CHECK: define internal noundef i32 [[TASK_CANCEL]](
 186 // CHECK: [[RES:%.+]] = call i32 @__kmpc_cancel(ptr @{{.+}}, i32 %{{.+}}, i32 4)
 187 // CHECK: [[IS_CANCEL:%.+]] = icmp ne i32 [[RES]], 0
 188 // CHECK: br i1 [[IS_CANCEL]], label %[[EXIT:.+]], label %[[CONTINUE:[^,]+]]
 189 // CHECK: [[EXIT]]:
 190 // CHECK: store i32 1, ptr [[CLEANUP_SLOT:%.+]],
 191 // CHECK: br label %[[DONE:[^,]+]]
 192 // CHECK: [[CONTINUE]]:
 193 // CHECK: [[RES:%.+]] = call i32 @__kmpc_cancellationpoint(ptr @{{.+}}, i32 %{{.+}}, i32 4)
 194 // CHECK: [[IS_CANCEL:%.+]] = icmp ne i32 [[RES]], 0
 195 // CHECK: br i1 [[IS_CANCEL]], label %[[EXIT2:.+]], label %[[CONTINUE2:[^,]+]]
 196 // CHECK: [[EXIT2]]:
 197 // CHECK: store i32 1, ptr [[CLEANUP_SLOT]],
 198 // CHECK: br label %[[DONE]]
 199 // CHECK: store i32 0, ptr [[CLEANUP_SLOT]],
 200 // CHECK: br label %[[DONE]]
 201 // CHECK: [[DONE]]:
 202 // CHECK: ret i32 0
 203
 204 // CHECK-LABEL: @_ZN1SC2Ei
 205 struct S {
 206   int a;
 207   S(int c) {
 208 // CHECK: [[GTID:%.+]] = call i32 @__kmpc_global_thread_num(ptr [[DEFLOC:@.+]])
 209 // CHECK: [[TASKV:%.+]] = call ptr @__kmpc_omp_task_alloc(ptr [[DEFLOC]], i32 [[GTID]], i32 1, i64 80, i64 16, ptr [[TASK4:@.+]])
 210 // CHECK: [[TASK_DATA:%.+]] = getelementptr inbounds nuw %{{.+}}, ptr [[TASKV]], i32 0, i32 0
 211 // CHECK: [[DOWN:%.+]] = getelementptr inbounds nuw [[TD_TY:%.+]], ptr [[TASK_DATA]], i32 0, i32 5
 212 // CHECK: store i64 0, ptr [[DOWN]],
 213 // CHECK: [[UP:%.+]] = getelementptr inbounds nuw [[TD_TY]], ptr [[TASK_DATA]], i32 0, i32 6
 214 // CHECK: store i64 %{{.+}}, ptr [[UP]],
 215 // CHECK: [[ST:%.+]] = getelementptr inbounds nuw [[TD_TY]], ptr [[TASK_DATA]], i32 0, i32 7
 216 // CHECK: store i64 1, ptr [[ST]],
 217 // CHECK: [[ST_VAL:%.+]] = load i64, ptr [[ST]],
 218 // CHECK: [[NUM_TASKS:%.+]] = zext i32 %{{.+}} to i64
 219 // CHECK: call void @__kmpc_taskloop(ptr [[DEFLOC]], i32 [[GTID]], ptr [[TASKV]], i32 1, ptr [[DOWN]], ptr [[UP]], i64 [[ST_VAL]], i32 1, i32 2, i64 [[NUM_TASKS]], ptr null)
 220 #pragma omp master taskloop shared(c) num_tasks(a)
 221     for (a = 0; a < c; ++a)
 222       ;
 223   }
 224 } s(1);
 225
 226 // CHECK: define internal noundef i32 [[TASK4]](
 227 // CHECK: [[DOWN:%.+]] = getelementptr inbounds nuw [[TD_TY:%.+]], ptr %{{.+}}, i32 0, i32 5
 228 // CHECK: [[DOWN_VAL:%.+]] = load i64, ptr [[DOWN]],
 229 // CHECK: [[UP:%.+]] = getelementptr inbounds nuw [[TD_TY]], ptr %{{.+}}, i32 0, i32 6
 230 // CHECK: [[UP_VAL:%.+]] = load i64, ptr [[UP]],
 231 // CHECK: [[ST:%.+]] = getelementptr inbounds nuw [[TD_TY]], ptr %{{.+}}, i32 0, i32 7
 232 // CHECK: [[ST_VAL:%.+]] = load i64, ptr [[ST]],
 233 // CHECK: [[LITER:%.+]] = getelementptr inbounds nuw [[TD_TY]], ptr %{{.+}}, i32 0, i32 8
 234 // CHECK: [[LITER_VAL:%.+]] = load i32, ptr [[LITER]],
 235 // CHECK: store i64 [[DOWN_VAL]], ptr [[LB:%[^,]+]],
 236 // CHECK: store i64 [[UP_VAL]], ptr [[UB:%[^,]+]],
 237 // CHECK: store i64 [[ST_VAL]], ptr [[ST:%[^,]+]],
 238 // CHECK: store i32 [[LITER_VAL]], ptr [[LITER:%[^,]+]],
 239 // CHECK: [[LB_VAL:%.+]] = load i64, ptr [[LB]],
 240 // CHECK: [[LB_I32:%.+]] = trunc i64 [[LB_VAL]] to i32
 241 // CHECK: store i32 [[LB_I32]], ptr [[CNT:%.+]],
 242 // CHECK: br label
 243 // CHECK: [[VAL:%.+]] = load i32, ptr [[CNT]],
 244 // CHECK: [[VAL_I64:%.+]] = sext i32 [[VAL]] to i64
 245 // CHECK: [[UB_VAL:%.+]] = load i64, ptr [[UB]],
 246 // CHECK: [[CMP:%.+]] = icmp ule i64 [[VAL_I64]], [[UB_VAL]]
 247 // CHECK: br i1 [[CMP]], label %{{.+}}, label %{{.+}}
 248 // CHECK: load i32, ptr %
 249 // CHECK: store i32 %
 250 // CHECK: load i32, ptr %
 251 // CHECK: add nsw i32 %{{.+}}, 1
 252 // CHECK: store i32 %{{.+}}, ptr %
 253 // CHECK: br label %
 254 // CHECK: ret i32 0
 255
 256 #endif