test/OpenMP/cancel_codegen.cpp

   1 // RUN: %clang_cc1 -verify -fopenmp -triple x86_64-apple-darwin13.4.0 -emit-llvm -o - %s | FileCheck %s
   2 // RUN: %clang_cc1 -fopenmp -x c++ -std=c++11 -triple x86_64-apple-darwin13.4.0 -emit-pch -o %t %s
   3 // RUN: %clang_cc1 -fopenmp -std=c++11 -include-pch %t -fsyntax-only -verify %s -triple x86_64-apple-darwin13.4.0 -emit-llvm -o - | FileCheck %s
   4 // expected-no-diagnostics
   5 // REQUIRES: x86-registered-target
   6 #ifndef HEADER
   7 #define HEADER
   8
   9 float flag;
  10 int main (int argc, char **argv) {
  11 // CHECK: [[GTID:%.+]] = call i32 @__kmpc_global_thread_num(
  12 #pragma omp parallel
  13 {
  14 #pragma omp cancel parallel if(flag)
  15   argv[0][0] = argc;
  16 }
  17 // CHECK: call void (%ident_t*, i32, void (i32*, i32*, ...)*, ...) @__kmpc_fork_call(
  18 #pragma omp sections
  19 {
  20 #pragma omp cancel sections
  21 }
  22 // CHECK: call void @__kmpc_for_static_init_4(
  23 // CHECK: call i32 @__kmpc_cancel(
  24 // CHECK: call i32 @__kmpc_cancel_barrier(%ident_t*
  25 // CHECK: call void @__kmpc_for_static_fini(
  26 // CHECK: call void @__kmpc_barrier(%ident_t*
  27 #pragma omp sections
  28 {
  29 #pragma omp cancel sections
  30 #pragma omp section
  31   {
  32 #pragma omp cancel sections
  33   }
  34 }
  35 // CHECK: call void @__kmpc_for_static_init_4(
  36 // CHECK: [[RES:%.+]] = call i32 @__kmpc_cancel(%ident_t* {{[^,]+}}, i32 [[GTID]], i32 3)
  37 // CHECK: [[CMP:%.+]] = icmp ne i32 [[RES]], 0
  38 // CHECK: br i1 [[CMP]], label %[[EXIT:[^,].+]], label %[[CONTINUE:.+]]
  39 // CHECK: [[EXIT]]
  40 // CHECK: call i32 @__kmpc_cancel_barrier(%ident_t*
  41 // CHECK: br label
  42 // CHECK: [[CONTINUE]]
  43 // CHECK: br label
  44 // CHECK: [[RES:%.+]] = call i32 @__kmpc_cancel(%ident_t* {{[^,]+}}, i32 [[GTID]], i32 3)
  45 // CHECK: [[CMP:%.+]] = icmp ne i32 [[RES]], 0
  46 // CHECK: br i1 [[CMP]], label %[[EXIT:[^,].+]], label %[[CONTINUE:.+]]
  47 // CHECK: [[EXIT]]
  48 // CHECK: call i32 @__kmpc_cancel_barrier(%ident_t*
  49 // CHECK: br label
  50 // CHECK: [[CONTINUE]]
  51 // CHECK: br label
  52 // CHECK: call void @__kmpc_for_static_fini(
  53 #pragma omp for
  54 for (int i = 0; i < argc; ++i) {
  55 #pragma omp cancel for if(cancel: flag)
  56 }
  57 // CHECK: call void @__kmpc_for_static_init_4(
  58 // CHECK: [[FLAG:%.+]] = load float, float* @{{.+}},
  59 // CHECK: [[BOOL:%.+]] = fcmp une float [[FLAG]], 0.000000e+00
  60 // CHECK: br i1 [[BOOL]], label %[[THEN:[^,]+]], label %[[ELSE:[^,]+]]
  61 // CHECK: [[THEN]]
  62 // CHECK: [[RES:%.+]] = call i32 @__kmpc_cancel(%ident_t* {{[^,]+}}, i32 [[GTID]], i32 2)
  63 // CHECK: [[CMP:%.+]] = icmp ne i32 [[RES]], 0
  64 // CHECK: br i1 [[CMP]], label %[[EXIT:[^,].+]], label %[[CONTINUE:.+]]
  65 // CHECK: [[EXIT]]
  66 // CHECK: call i32 @__kmpc_cancel_barrier(%ident_t*
  67 // CHECK: br label
  68 // CHECK: [[CONTINUE]]
  69 // CHECK: br label
  70 // CHECK: [[ELSE]]
  71 // CHECK: br label
  72 // CHECK: call void @__kmpc_for_static_fini(
  73 // CHECK: call void @__kmpc_barrier(%ident_t*
  74 #pragma omp task
  75 {
  76 #pragma omp cancel taskgroup
  77 }
  78 // CHECK: call i8* @__kmpc_omp_task_alloc(
  79 // CHECK: call i32 @__kmpc_omp_task(
  80 #pragma omp parallel sections
  81 {
  82 #pragma omp cancel sections
  83 }
  84 // CHECK: call void (%ident_t*, i32, void (i32*, i32*, ...)*, ...) @__kmpc_fork_call(
  85 #pragma omp parallel sections
  86 {
  87 #pragma omp cancel sections
  88 #pragma omp section
  89   {
  90 #pragma omp cancel sections
  91   }
  92 }
  93 // CHECK: call void (%ident_t*, i32, void (i32*, i32*, ...)*, ...) @__kmpc_fork_call(
  94 int r = 0;
  95 #pragma omp parallel for reduction(+: r)
  96 for (int i = 0; i < argc; ++i) {
  97 #pragma omp cancel for
  98   r += i;
  99 }
 100 // CHECK: call void (%ident_t*, i32, void (i32*, i32*, ...)*, ...) @__kmpc_fork_call(
 101   return argc;
 102 }
 103
 104 // CHECK: define internal void @{{[^(]+}}(i32* {{[^,]+}}, i32* {{[^,]+}},
 105 // CHECK: [[FLAG:%.+]] = load float, float* @{{.+}},
 106 // CHECK: [[BOOL:%.+]] = fcmp une float [[FLAG]], 0.000000e+00
 107 // CHECK: br i1 [[BOOL]], label %[[THEN:[^,]+]], label %[[ELSE:[^,]+]]
 108 // CHECK: [[THEN]]
 109 // CHECK: [[RES:%.+]] = call i32 @__kmpc_cancel(%ident_t* {{[^,]+}}, i32 {{[^,]+}}, i32 1)
 110 // CHECK: [[CMP:%.+]] = icmp ne i32 [[RES]], 0
 111 // CHECK: br i1 [[CMP]], label %[[EXIT:[^,]+]],
 112 // CHECK: [[EXIT]]
 113 // CHECK: call i32 @__kmpc_cancel_barrier(%ident_t*
 114 // CHECK: br label %[[RETURN:.+]]
 115 // CHECK: [[ELSE]]
 116 // CHECK: br label
 117 // CHECK: [[RETURN]]
 118 // CHECK: ret void
 119
 120 // CHECK: define internal i32 @{{[^(]+}}(i32
 121 // CHECK: [[RES:%.+]] = call i32 @__kmpc_cancel(%ident_t* {{[^,]+}}, i32 {{[^,]+}}, i32 4)
 122 // CHECK: [[CMP:%.+]] = icmp ne i32 [[RES]], 0
 123 // CHECK: br i1 [[CMP]], label %[[EXIT:[^,]+]],
 124 // CHECK: [[EXIT]]
 125 // CHECK: call i32 @__kmpc_cancel_barrier(%ident_t*
 126 // CHECK: br label %[[RETURN:.+]]
 127 // CHECK: [[RETURN]]
 128 // CHECK: ret i32 0
 129
 130 // CHECK: define internal void @{{[^(]+}}(i32* {{[^,]+}}, i32* {{[^,]+}})
 131 // CHECK: call void @__kmpc_for_static_init_4(
 132 // CHECK: call i32 @__kmpc_cancel(
 133 // CHECK: call i32 @__kmpc_cancel_barrier(%ident_t*
 134 // CHECK: call void @__kmpc_for_static_fini(
 135 // CHECK: ret void
 136
 137 // CHECK: define internal void @{{[^(]+}}(i32* {{[^,]+}}, i32* {{[^,]+}})
 138 // CHECK: call void @__kmpc_for_static_init_4(
 139 // CHECK: [[RES:%.+]] = call i32 @__kmpc_cancel(%ident_t* {{[^,]+}}, i32 [[GTID:%.+]], i32 3)
 140 // CHECK: [[CMP:%.+]] = icmp ne i32 [[RES]], 0
 141 // CHECK: br i1 [[CMP]], label %[[EXIT:[^,].+]], label %[[CONTINUE:.+]]
 142 // CHECK: [[EXIT]]
 143 // CHECK: call i32 @__kmpc_cancel_barrier(%ident_t*
 144 // CHECK: br label
 145 // CHECK: [[CONTINUE]]
 146 // CHECK: br label
 147 // CHECK: [[RES:%.+]] = call i32 @__kmpc_cancel(%ident_t* {{[^,]+}}, i32 [[GTID]], i32 3)
 148 // CHECK: [[CMP:%.+]] = icmp ne i32 [[RES]], 0
 149 // CHECK: br i1 [[CMP]], label %[[EXIT:[^,].+]], label %[[CONTINUE:.+]]
 150 // CHECK: [[EXIT]]
 151 // CHECK: call i32 @__kmpc_cancel_barrier(%ident_t*
 152 // CHECK: br label
 153 // CHECK: [[CONTINUE]]
 154 // CHECK: br label
 155 // CHECK: call void @__kmpc_for_static_fini(
 156 // CHECK: ret void
 157
 158 // CHECK: define internal void @{{[^(]+}}(i32* {{[^,]+}}, i32* {{[^,]+}},
 159 // CHECK: call void @__kmpc_for_static_init_4(
 160 // CHECK: [[RES:%.+]] = call i32 @__kmpc_cancel(%ident_t* {{[^,]+}}, i32 [[GTID:%.+]], i32 2)
 161 // CHECK: [[CMP:%.+]] = icmp ne i32 [[RES]], 0
 162 // CHECK: br i1 [[CMP]], label %[[EXIT:[^,].+]], label %[[CONTINUE:.+]]
 163 // CHECK: [[EXIT]]
 164 // CHECK: call i32 @__kmpc_cancel_barrier(%ident_t*
 165 // CHECK: br label
 166 // CHECK: [[CONTINUE]]
 167 // CHECK: br label
 168 // CHECK: call void @__kmpc_for_static_fini(
 169 // CHECK: call i32 @__kmpc_reduce_nowait(
 170 // CHECK: call void @__kmpc_end_reduce_nowait(
 171 // CHECK: call void @__kmpc_for_static_fini(
 172 // CHECK: ret void
 173
 174 #endif