clang/test/OpenMP/critical_codegen_attr.cpp

   1 // RUN: %clang_cc1 -verify -fopenmp -emit-llvm %s -fexceptions -fcxx-exceptions -o - | FileCheck %s --check-prefixes=ALL,NORMAL
   2 // RUN: %clang_cc1 -fopenmp -std=c++11 -triple x86_64-unknown-unknown -fexceptions -fcxx-exceptions -emit-pch -o %t %s
   3 // RUN: %clang_cc1 -fopenmp -triple x86_64-unknown-unknown -fexceptions -fcxx-exceptions -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck %s --check-prefixes=ALL,NORMAL
   4 // RUN: %clang_cc1 -verify -triple x86_64-apple-darwin10 -fopenmp -fexceptions -fcxx-exceptions -debug-info-kind=line-tables-only -emit-llvm %s -o - | FileCheck %s --check-prefix=TERM_DEBUG
   5 // RUN: %clang_cc1 -verify -fopenmp -fopenmp-enable-irbuilder -emit-llvm %s -fexceptions -fcxx-exceptions -o - | FileCheck %s --check-prefixes=ALL,IRBUILDER
   6 // RUN: %clang_cc1 -fopenmp -fopenmp-enable-irbuilder -std=c++11 -triple x86_64-unknown-unknown -fexceptions -fcxx-exceptions -emit-pch -o %t %s
   7 // RUN: %clang_cc1 -fopenmp -fopenmp-enable-irbuilder -triple x86_64-unknown-unknown -fexceptions -fcxx-exceptions -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck %s --check-prefixes=ALL,IRBUILDER
   8
   9 // RUN: %clang_cc1 -verify -fopenmp-simd -emit-llvm %s -fexceptions -fcxx-exceptions -o - | FileCheck --check-prefix SIMD-ONLY0 %s
  10 // RUN: %clang_cc1 -fopenmp-simd -std=c++11 -triple x86_64-unknown-unknown -fexceptions -fcxx-exceptions -emit-pch -o %t %s
  11 // RUN: %clang_cc1 -fopenmp-simd -triple x86_64-unknown-unknown -fexceptions -fcxx-exceptions -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck --check-prefix SIMD-ONLY0 %s
  12 // RUN: %clang_cc1 -verify -triple x86_64-apple-darwin10 -fopenmp-simd -fexceptions -fcxx-exceptions -debug-info-kind=line-tables-only -emit-llvm %s -o - | FileCheck --check-prefix SIMD-ONLY0 %s
  13 // SIMD-ONLY0-NOT: {{__kmpc|__tgt}}
  14 // expected-no-diagnostics
  15 #ifndef HEADER
  16 #define HEADER
  17
  18 // ALL:       [[IDENT_T_TY:%.+]] = type { i32, i32, i32, i32, ptr }
  19 // ALL:       [[UNNAMED_LOCK:@.+]] = common global [8 x i32] zeroinitializer
  20 // ALL:       [[THE_NAME_LOCK:@.+]] = common global [8 x i32] zeroinitializer
  21 // ALL:       [[THE_NAME_LOCK1:@.+]] = common global [8 x i32] zeroinitializer
  22
  23 // ALL:       define {{.*}}void [[FOO:@.+]]()
  24
  25 void foo() { extern void mayThrow(); mayThrow(); }
  26
  27 // ALL-LABEL: @main
  28 // TERM_DEBUG-LABEL: @main
  29 int main() {
  30   // ALL:       [[A_ADDR:%.+]] = alloca i8
  31   char a;
  32
  33 // ALL:                         [[GTID:%.+]] = call {{.*}}i32 @__kmpc_global_thread_num(ptr [[DEFAULT_LOC:@.+]])
  34 // ALL:                         call {{.*}}void @__kmpc_critical(ptr [[DEFAULT_LOC]], i32 [[GTID]], ptr [[UNNAMED_LOCK]])
  35 // ALL-NEXT:                    store i8 2, ptr [[A_ADDR]]
  36 // IRBUILDER-NEXT:              br label %[[AFTER:[^ ,]+]]
  37 // IRBUILDER:                   [[AFTER]]
  38 // ALL-NEXT:                    call {{.*}}void @__kmpc_end_critical(ptr [[DEFAULT_LOC]], i32 [[GTID]], ptr [[UNNAMED_LOCK]])
  39   [[omp::directive(critical)]]
  40   a = 2;
  41 // IRBUILDER:       [[GTID:%.+]] = call {{.*}}i32 @__kmpc_global_thread_num(ptr [[DEFAULT_LOC:@.+]])
  42 // ALL:                         call {{.*}}void @__kmpc_critical(ptr [[DEFAULT_LOC]], i32 [[GTID]], ptr [[THE_NAME_LOCK]])
  43 // IRBUILDER-NEXT:      call {{.*}}void [[FOO]]()
  44 // NORMAL-NEXT:                 invoke {{.*}}void [[FOO]]()
  45 // IRBUILDER-NEXT:              br label %[[AFTER:[^ ,]+]]
  46 // IRBUILDER:                   [[AFTER]]
  47 // ALL:                                 call {{.*}}void @__kmpc_end_critical(ptr [[DEFAULT_LOC]], i32 [[GTID]], ptr [[THE_NAME_LOCK]])
  48   [[omp::directive(critical(the_name))]]
  49   foo();
  50 // IRBUILDER:                   [[GTID:%.+]] = call {{.*}}i32 @__kmpc_global_thread_num(ptr [[DEFAULT_LOC:@.+]])
  51 // ALL:                         call {{.*}}void @__kmpc_critical_with_hint(ptr [[DEFAULT_LOC]], i32 [[GTID]], ptr [[THE_NAME_LOCK1]], i{{64|32}} 23)
  52 // IRBUILDER-NEXT:      call {{.*}}void [[FOO]]()
  53 // NORMAL-NEXT:           invoke {{.*}}void [[FOO]]()
  54 // IRBUILDER-NEXT:              br label %[[AFTER:[^ ,]+]]
  55 // IRBUILDER:                   [[AFTER]]
  56 // ALL:                         call {{.*}}void @__kmpc_end_critical(ptr [[DEFAULT_LOC]], i32 [[GTID]], ptr [[THE_NAME_LOCK1]])
  57   [[omp::directive(critical(the_name1) hint(23))]]
  58   foo();
  59   // IRBUILDER:                 [[GTID:%.+]] = call {{.*}}i32 @__kmpc_global_thread_num(ptr [[DEFAULT_LOC:@.+]])
  60   // ALL:       call {{.*}}void @__kmpc_critical(ptr [[DEFAULT_LOC]], i32 [[GTID]], ptr [[THE_NAME_LOCK]])
  61   // NORMAL:       br label
  62   // NORMAL-NOT:   call {{.*}}void @__kmpc_end_critical(
  63   // NORMAL:       br label
  64   // NORMAL-NOT:   call {{.*}}void @__kmpc_end_critical(
  65   // NORMAL:       br label
  66   if (a)
  67     [[omp::directive(critical(the_name))]]
  68     while (1)
  69       ;
  70   // ALL:  call {{.*}}void [[FOO]]()
  71   foo();
  72   // ALL-NOT:   call void @__kmpc_critical
  73   // ALL-NOT:   call void @__kmpc_end_critical
  74   return a;
  75 }
  76
  77 // ALL-LABEL:        lambda_critical
  78 // TERM_DEBUG-LABEL: lambda_critical
  79 void lambda_critical(int a, int b) {
  80   auto l = [=]() {
  81     [[omp::directive(critical)]]
  82     {
  83       // ALL: call void @__kmpc_critical(
  84       int c = a + b;
  85     }
  86   };
  87
  88   l();
  89
  90   auto l1 = [=]() {
  91     [[omp::sequence(directive(parallel), directive(critical))]]
  92     {
  93       // ALL: call void @__kmpc_critical(
  94       int c = a + b;
  95     }
  96   };
  97
  98   l1();
  99 }
 100
 101 struct S {
 102   int a;
 103 };
 104 // ALL-LABEL: critical_ref
 105 void critical_ref(S &s) {
 106   // ALL: [[S_ADDR:%.+]] = alloca ptr,
 107   // ALL: [[S_REF:%.+]] = load ptr, ptr [[S_ADDR]],
 108   // ALL: [[S_A_REF:%.+]] = getelementptr inbounds %struct.S, ptr [[S_REF]], i32 0, i32 0
 109   ++s.a;
 110   // ALL: call void @__kmpc_critical(
 111   [[omp::directive(critical)]]
 112   // ALL: [[S_REF:%.+]] = load ptr, ptr [[S_ADDR]],
 113   // ALL: [[S_A_REF:%.+]] = getelementptr inbounds %struct.S, ptr [[S_REF]], i32 0, i32 0
 114   ++s.a;
 115   // ALL: call void @__kmpc_end_critical(
 116 }
 117
 118 // ALL-LABEL:      parallel_critical
 119 // TERM_DEBUG-LABEL: parallel_critical
 120 void parallel_critical() {
 121   [[omp::sequence(directive(parallel), directive(critical))]]
 122   // TERM_DEBUG-NOT: __kmpc_global_thread_num
 123   // TERM_DEBUG:     call void @__kmpc_critical({{.+}}), !dbg [[DBG_LOC_START:![0-9]+]]
 124   // TERM_DEBUG:     invoke void {{.*}}foo{{.*}}()
 125   // TERM_DEBUG:     unwind label %[[TERM_LPAD:.+]],
 126   // TERM_DEBUG-NOT: __kmpc_global_thread_num
 127   // TERM_DEBUG:     call void @__kmpc_end_critical({{.+}}), !dbg [[DBG_LOC_END:![0-9]+]]
 128   // TERM_DEBUG:     [[TERM_LPAD]]
 129   // TERM_DEBUG:     call void @__clang_call_terminate
 130   // TERM_DEBUG:     unreachable
 131   foo();
 132 }
 133 // TERM_DEBUG-DAG: [[DBG_LOC_START]] = !DILocation(line: [[@LINE-12]],
 134 // TERM_DEBUG-DAG: [[DBG_LOC_END]] = !DILocation(line: [[@LINE-3]],
 135 #endif
 136