1 // Test target codegen - host bc file has to be created first.
2 // RUN: %clang_cc1 -verify -fopenmp -x c++ -triple powerpc64le-unknown-unknown -fopenmp-targets=nvptx64-nvidia-cuda -emit-llvm-bc %s -o %t-ppc-host.bc
3 // RUN: %clang_cc1 -verify -fopenmp -x c++ -triple nvptx64-unknown-unknown -fopenmp-targets=nvptx64-nvidia-cuda -emit-llvm %s -fopenmp-is-device -fopenmp-host-ir-file-path %t-ppc-host.bc -o - | FileCheck %s --check-prefix CHECK --check-prefix CHECK-64
4 // expected-no-diagnostics
5 #ifndef HEADER
6 #define HEADER
7 
8 template<typename tx>
9 tx ftemplate(int n) {
10   tx b[10];
11 
12   #pragma omp target
13   {
14     tx d = n;
15     #pragma omp parallel for
16     for(int i=0; i<10; i++) {
17       b[i] += d;
18     }
19     b[3] += 1;
20   }
21 
22   return b[3];
23 }
24 
25 int bar(int n){
26   int a = 0;
27 
28   a += ftemplate<int>(n);
29 
30   return a;
31 }
32 
33 // CHECK-LABEL: define {{.*}}void {{@__omp_offloading_.+template.+l12}}_worker()
34 // CHECK: call void @llvm.nvvm.barrier0()
35 // CHECK: call i1 @__kmpc_kernel_parallel(
36 // CHECK: call void @__omp_outlined___wrapper(
37 
38 // CHECK: define weak void @__omp_offloading_{{.*}}l12(
39 // CHECK: call void @__omp_offloading_{{.*}}l12_worker()
40 // CHECK: call void @__kmpc_kernel_init(
41 // CHECK: call void @__kmpc_data_sharing_init_stack()
42 // CHECK: call i8* @__kmpc_data_sharing_push_stack(i64 4, i16 0)
43 // CHECK: call void @__kmpc_kernel_prepare_parallel(
44 // CHECK: call void @__kmpc_begin_sharing_variables({{.*}}, i64 2)
45 // CHECK: call void @llvm.nvvm.barrier0()
46 // CHECK: call void @llvm.nvvm.barrier0()
47 // CHECK: call void @__kmpc_end_sharing_variables()
48 // CHECK: call void @__kmpc_data_sharing_pop_stack(
49 // CHECK: call void @__kmpc_kernel_deinit(i16 1)
50 
51 // CHECK: define internal void @__omp_outlined__(
52 // CHECK: alloca
53 // CHECK: alloca
54 // CHECK: alloca
55 // CHECK: alloca
56 // CHECK: [[OMP_IV:%.*]] = alloca i32
57 // CHECK: store i32 0, {{.*}} [[OMP_LB:%.+]],
58 // CHECK: store i32 9, {{.*}} [[OMP_UB:%.+]],
59 // CHECK: store i32 1, {{.*}} [[OMP_ST:%.+]],
60 // CHECK: call void @__kmpc_for_static_init_4({{.*}} i32 33, {{.*}} [[OMP_LB]], {{.*}} [[OMP_UB]], {{.*}} [[OMP_ST]], i32 1, i32 1)
61 // CHECK: br label %[[OMP_DISPATCH_COND:.+]]
62 
63 // CHECK: [[OMP_DISPATCH_COND]]
64 // CHECK: [[OMP_UB_1:%.+]] = load {{.*}} [[OMP_UB]]
65 // CHECK: [[COMP_1:%.+]] = icmp sgt {{.*}} [[OMP_UB_1]]
66 // CHECK: br i1 [[COMP_1]], label %[[COND_TRUE:.+]], label %[[COND_FALSE:.+]]
67 
68 // CHECK: [[COND_TRUE]]
69 // CHECK: br label %[[COND_END:.+]]
70 
71 // CHECK: [[COND_FALSE]]
72 // CHECK: [[OMP_UB_2:%.+]] = load {{.*}}* [[OMP_UB]]
73 // CHECK: br label %[[COND_END]]
74 
75 // CHECK: [[COND_END]]
76 // CHECK: [[COND_RES:%.+]] = phi i32 [ 9, %[[COND_TRUE]] ], [ [[OMP_UB_2]], %[[COND_FALSE]] ]
77 // CHECK: store i32 [[COND_RES]], i32* [[OMP_UB]]
78 // CHECK: [[OMP_LB_1:%.+]] = load i32, i32* [[OMP_LB]]
79 // CHECK: store i32 [[OMP_LB_1]], i32* [[OMP_IV]]
80 // CHECK: [[OMP_IV_1:%.+]] = load i32, i32* [[OMP_IV]]
81 // CHECK: [[OMP_UB_3:%.+]] = load i32, i32* [[OMP_UB]]
82 // CHECK: [[COMP_2:%.+]] = icmp sle i32 [[OMP_IV_1]], [[OMP_UB_3]]
83 // CHECK: br i1 [[COMP_2]], label %[[DISPATCH_BODY:.+]], label %[[DISPATCH_END:.+]]
84 
85 // CHECK: [[DISPATCH_BODY]]
86 // CHECK: br label %[[OMP_INNER_FOR_COND:.+]]
87 
88 // CHECK: [[OMP_INNER_FOR_COND]]
89 // CHECK: [[OMP_IV_2:%.+]] = load i32, i32* [[OMP_IV]]
90 // CHECK: [[OMP_UB_4:%.+]] = load i32, i32* [[OMP_UB]]
91 // CHECK: [[COMP_3:%.+]] = icmp sle i32 [[OMP_IV_2]], [[OMP_UB_4]]
92 // CHECK: br i1 [[COMP_3]], label %[[OMP_INNER_FOR_BODY:.+]], label %[[OMP_INNER_FOR_END:.+]]
93 
94 // CHECK: [[OMP_INNER_FOR_BODY]]
95 // CHECK: br label %[[OMP_BODY_CONTINUE:.+]]
96 
97 // CHECK: [[OMP_BODY_CONTINUE]]
98 // CHECK: br label %[[OMP_INNER_FOR_INC:.+]]
99 
100 // CHECK: [[OMP_INNER_FOR_INC]]
101 // CHECK: [[OMP_IV_3:%.+]] = load i32, i32* [[OMP_IV]]
102 // CHECK: [[ADD_1:%.+]] = add nsw i32 [[OMP_IV_3]], 1
103 // CHECK: store i32 [[ADD_1]], i32* [[OMP_IV]]
104 // CHECK: br label %[[OMP_INNER_FOR_COND]]
105 
106 // CHECK: [[OMP_INNER_FOR_COND]]
107 // CHECK: br label %[[OMP_DISPATCH_INC:.+]]
108 
109 // CHECK: [[OMP_DISPATCH_INC]]
110 // CHECK: [[OMP_LB_2:%.+]] = load i32, i32* [[OMP_LB]]
111 // CHECK: [[OMP_ST_1:%.+]] = load i32, i32* [[OMP_ST]]
112 // CHECK: [[ADD_2:%.+]] = add nsw i32 [[OMP_LB_2]], [[OMP_ST_1]]
113 // CHECK: store i32 [[ADD_2]], i32* [[OMP_LB]]
114 // CHECK: [[OMP_UB_5:%.+]] = load i32, i32* [[OMP_UB]]
115 // CHECK: [[OMP_ST_2:%.+]] = load i32, i32* [[OMP_ST]]
116 // CHECK: [[ADD_3:%.+]] = add nsw i32 [[OMP_UB_5]], [[OMP_ST_2]]
117 // CHECK: store i32 [[ADD_3]], i32* [[OMP_UB]]
118 
119 // CHECK: [[DISPATCH_END]]
120 // CHECK: call void @__kmpc_for_static_fini(
121 // CHECK: ret void
122 
123 #endif
124