test/OpenMP/ordered_doacross_codegen.c

   1 // RUN: %clang_cc1 -verify -fopenmp -triple x86_64-unknown-unknown -emit-llvm %s -o - | FileCheck %s
   2 // RUN: %clang_cc1 -fopenmp -triple x86_64-unknown-unknown -emit-pch -o %t %s
   3 // RUN: %clang_cc1 -fopenmp -triple x86_64-unknown-unknown -include-pch %t -verify %s -emit-llvm -o - | FileCheck %s
   4
   5 // RUN: %clang_cc1 -verify -fopenmp-simd -triple x86_64-unknown-unknown -emit-llvm %s -o - | FileCheck --check-prefix SIMD-ONLY0 %s
   6 // RUN: %clang_cc1 -fopenmp-simd -triple x86_64-unknown-unknown -emit-pch -o %t %s
   7 // RUN: %clang_cc1 -fopenmp-simd -triple x86_64-unknown-unknown -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
  11 #ifndef HEADER
  12 #define HEADER
  13
  14 // CHECK: [[KMP_DIM:%.+]] = type { i64, i64, i64 }
  15 extern int n;
  16 int a[10], b[10], c[10], d[10];
  17 void foo();
  18
  19 // CHECK-LABEL: @main()
  20 int main() {
  21   int i;
  22 // CHECK: [[DIMS:%.+]] = alloca [[KMP_DIM]],
  23 // CHECK: [[GTID:%.+]] = call i32 @__kmpc_global_thread_num([[IDENT:%.+]])
  24 // CHECK: icmp
  25 // CHECK-NEXT: br i1 %
  26 // CHECK: [[CAST:%.+]] = bitcast [[KMP_DIM]]* [[DIMS]] to i8*
  27 // CHECK: call void @llvm.memset.p0i8.i64(i8* align 8 [[CAST]], i8 0, i64 24, i1 false)
  28 // CHECK: getelementptr inbounds [[KMP_DIM]], [[KMP_DIM]]* [[DIMS]], i32 0, i32 1
  29 // CHECK: store i64 %{{.+}}, i64* %
  30 // CHECK: getelementptr inbounds [[KMP_DIM]], [[KMP_DIM]]* [[DIMS]], i32 0, i32 2
  31 // CHECK: store i64 1, i64* %
  32 // CHECK: [[CAST:%.+]] = bitcast [[KMP_DIM]]* [[DIMS]] to i8*
  33 // CHECK: call void @__kmpc_doacross_init([[IDENT]], i32 [[GTID]], i32 1, i8* [[CAST]])
  34 // CHECK: call void @__kmpc_for_static_init_4(
  35 #pragma omp for ordered(1)
  36   for (i = 0; i < n; ++i) {
  37     a[i] = b[i] + 1;
  38     foo();
  39 // CHECK: call void [[FOO:.+]](
  40 // CHECK: load i32, i32* [[CNT:%.+]],
  41 // CHECK-NEXT: sext i32 %{{.+}} to i64
  42 // CHECK-NEXT: store i64 %{{.+}}, i64* [[TMP:%.+]],
  43 // CHECK-NEXT: call void @__kmpc_doacross_post([[IDENT]], i32 [[GTID]], i64* [[TMP]])
  44 #pragma omp ordered depend(source)
  45     c[i] = c[i] + 1;
  46     foo();
  47 // CHECK: call void [[FOO]]
  48 // CHECK: load i32, i32* [[CNT]],
  49 // CHECK-NEXT: sub nsw i32 %{{.+}}, 2
  50 // CHECK-NEXT: sext i32 %{{.+}} to i64
  51 // CHECK-NEXT: store i64 %{{.+}}, i64* [[TMP:%.+]],
  52 // CHECK-NEXT: call void @__kmpc_doacross_wait([[IDENT]], i32 [[GTID]], i64* [[TMP]])
  53 #pragma omp ordered depend(sink : i - 2)
  54     d[i] = a[i - 2];
  55   }
  56   // CHECK: call void @__kmpc_for_static_fini(
  57   // CHECK: call void @__kmpc_doacross_fini([[IDENT]], i32 [[GTID]])
  58   // CHECK: ret i32 0
  59   return 0;
  60 }
  61 #endif // HEADER