Gitiles
Code Review Sign In
gerrit-public.fairphone.software / toolchain / llvm-project / a8a9153a372afcfe2d18399b079bfdc5fd4da626 / . / clang / test / OpenMP / teams_distribute_collapse_codegen.cpp
blob: 716e766929134a5862e8aa6fa5236b7caef42f7d [file] [log] [blame]
Carlo Bertolliba1487b2017-10-04 14:12:09 +0000[diff] [blame]1// expected-no-diagnostics
2#ifndef HEADER
3#define HEADER
4
5// Test host codegen.
6// RUN: %clang_cc1 -DCK1 -verify -fopenmp -x c++ -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -emit-llvm %s -o - | FileCheck %s --check-prefix CK1 --check-prefix CK1-64
7// RUN: %clang_cc1 -DCK1 -fopenmp -x c++ -std=c++11 -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -emit-pch -o %t %s
8// RUN: %clang_cc1 -DCK1 -fopenmp -x c++ -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck %s --check-prefix CK1 --check-prefix CK1-64
9// RUN: %clang_cc1 -DCK1 -verify -fopenmp -x c++ -triple i386-unknown-unknown -fopenmp-targets=i386-pc-linux-gnu -emit-llvm %s -o - | FileCheck %s --check-prefix CK1 --check-prefix CK1-32
10// RUN: %clang_cc1 -DCK1 -fopenmp -x c++ -std=c++11 -triple i386-unknown-unknown -fopenmp-targets=i386-pc-linux-gnu -emit-pch -o %t %s
11// RUN: %clang_cc1 -DCK1 -fopenmp -x c++ -triple i386-unknown-unknown -fopenmp-targets=i386-pc-linux-gnu -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck %s --check-prefix CK1 --check-prefix CK1-32
Alexey Bataeva8a9153a2017-12-29 18:07:07 +0000[diff] [blame^]12
13// RUN: %clang_cc1 -DCK1 -verify -fopenmp-simd -x c++ -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -emit-llvm %s -o - | FileCheck --check-prefix SIMD-ONLY0 %s
14// RUN: %clang_cc1 -DCK1 -fopenmp-simd -x c++ -std=c++11 -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -emit-pch -o %t %s
15// RUN: %clang_cc1 -DCK1 -fopenmp-simd -x c++ -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck --check-prefix SIMD-ONLY0 %s
16// RUN: %clang_cc1 -DCK1 -verify -fopenmp-simd -x c++ -triple i386-unknown-unknown -fopenmp-targets=i386-pc-linux-gnu -emit-llvm %s -o - | FileCheck --check-prefix SIMD-ONLY0 %s
17// RUN: %clang_cc1 -DCK1 -fopenmp-simd -x c++ -std=c++11 -triple i386-unknown-unknown -fopenmp-targets=i386-pc-linux-gnu -emit-pch -o %t %s
18// RUN: %clang_cc1 -DCK1 -fopenmp-simd -x c++ -triple i386-unknown-unknown -fopenmp-targets=i386-pc-linux-gnu -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck --check-prefix SIMD-ONLY0 %s
19// SIMD-ONLY0-NOT: {{__kmpc|__tgt}}
Carlo Bertolliba1487b2017-10-04 14:12:09 +0000[diff] [blame]20#ifdef CK1
21
22template <typename T, int X, long long Y>
23struct SS{
24 T a[X][Y];
25
26 // CK1: define {{.*}}i32 @{{.+}}foo{{.+}}(
27 int foo(void) {
28
Alexey Bataev50a1c782017-12-01 21:31:08 +0000[diff] [blame]29 // CK1: call i32 @__tgt_target_teams(
Carlo Bertolliba1487b2017-10-04 14:12:09 +0000[diff] [blame]30 // CK1: call void @[[OFFL1:.+]](
31 #pragma omp target
32 #pragma omp teams distribute collapse(2)
33 for(int i = 0; i < X; i++) {
34 for(int j = 0; j < Y; j++) {
35 a[i][j] = (T)0;
36 }
37 }
38 // CK1: define internal void @[[OFFL1]](
39 // CK1: call void {{.+}} @__kmpc_fork_teams({{.+}}, i32 1, {{.+}} @[[OUTL1:.+]] to {{.+}},
40 // CK1: ret void
41
42 // CK1: define internal void @[[OUTL1]]({{.+}})
43 // discard loop variables not needed here
44 // CK1: = alloca i32,
45 // CK1: = alloca i32,
Alexey Bataevc2e88a82017-12-04 21:30:42 +0000[diff] [blame]46 // CK1: = alloca i32,
47 // CK1: = alloca i32,
Carlo Bertolliba1487b2017-10-04 14:12:09 +0000[diff] [blame]48 // CK1: [[OMP_UB:%.+]] = alloca i32,
49 // CK1: store i32 56087, i32* [[OMP_UB]],
50 // CK1: call void @__kmpc_for_static_init_4({{.+}}, {{.+}}, i32 92, {{.+}}, {{.+}}, i32* [[OMP_UB]],
51 // CK1: call void @__kmpc_for_static_fini(
52 // CK1: ret void
53
54 return a[0][0];
55 }
56};
57
58int teams_template_struct(void) {
59 SS<int, 123, 456> V;
60 return V.foo();
61
62}
63#endif // CK1
64
65// Test host codegen.
66// RUN: %clang_cc1 -DCK2 -verify -fopenmp -x c++ -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -emit-llvm %s -o - | FileCheck %s --check-prefix CK2 --check-prefix CK2-64
67// RUN: %clang_cc1 -DCK2 -fopenmp -x c++ -std=c++11 -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -emit-pch -o %t %s
68// RUN: %clang_cc1 -DCK2 -fopenmp -x c++ -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck %s --check-prefix CK2 --check-prefix CK2-64
69// RUN: %clang_cc1 -DCK2 -verify -fopenmp -x c++ -triple i386-unknown-unknown -fopenmp-targets=i386-pc-linux-gnu -emit-llvm %s -o - | FileCheck %s --check-prefix CK2 --check-prefix CK2-32
70// RUN: %clang_cc1 -DCK2 -fopenmp -x c++ -std=c++11 -triple i386-unknown-unknown -fopenmp-targets=i386-pc-linux-gnu -emit-pch -o %t %s
71// RUN: %clang_cc1 -DCK2 -fopenmp -x c++ -triple i386-unknown-unknown -fopenmp-targets=i386-pc-linux-gnu -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck %s --check-prefix CK2 --check-prefix CK2-32
Alexey Bataeva8a9153a2017-12-29 18:07:07 +0000[diff] [blame^]72
73// RUN: %clang_cc1 -DCK2 -verify -fopenmp-simd -x c++ -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -emit-llvm %s -o - | FileCheck --check-prefix SIMD-ONLY1 %s
74// RUN: %clang_cc1 -DCK2 -fopenmp-simd -x c++ -std=c++11 -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -emit-pch -o %t %s
75// RUN: %clang_cc1 -DCK2 -fopenmp-simd -x c++ -triple powerpc64le-unknown-unknown -fopenmp-targets=powerpc64le-ibm-linux-gnu -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck --check-prefix SIMD-ONLY1 %s
76// RUN: %clang_cc1 -DCK2 -verify -fopenmp-simd -x c++ -triple i386-unknown-unknown -fopenmp-targets=i386-pc-linux-gnu -emit-llvm %s -o - | FileCheck --check-prefix SIMD-ONLY1 %s
77// RUN: %clang_cc1 -DCK2 -fopenmp-simd -x c++ -std=c++11 -triple i386-unknown-unknown -fopenmp-targets=i386-pc-linux-gnu -emit-pch -o %t %s
78// RUN: %clang_cc1 -DCK2 -fopenmp-simd -x c++ -triple i386-unknown-unknown -fopenmp-targets=i386-pc-linux-gnu -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck --check-prefix SIMD-ONLY1 %s
79// SIMD-ONLY1-NOT: {{__kmpc|__tgt}}
Carlo Bertolliba1487b2017-10-04 14:12:09 +0000[diff] [blame]80#ifdef CK2
81
82template <typename T, int n, int m>
83int tmain(T argc) {
84 T a[n][m];
85 #pragma omp target
86 #pragma omp teams distribute collapse(2)
87 for(int i = 0; i < n; i++) {
88 for(int j = 0; j < m; j++) {
89 a[i][j] = (T)0;
90 }
91 }
92 return 0;
93}
94
95int main (int argc, char **argv) {
96 int n = 100;
97 int m = 2;
98 int a[n][m];
99 #pragma omp target
100 #pragma omp teams distribute collapse(2)
101 for(int i = 0; i < n; i++) {
102 for(int j = 0; j < m; j++) {
103 a[i][j] = 0;
104 }
105 }
106 return tmain<int, 10, 2>(argc);
107}
108
109// CK2: define {{.*}}i32 @{{[^,]+}}(i{{.+}}{{.+}} %[[ARGC:.+]], {{.+}})
Alexey Bataev50a1c782017-12-01 21:31:08 +0000[diff] [blame]110// CK2: call i32 @__tgt_target_teams(
Carlo Bertolliba1487b2017-10-04 14:12:09 +0000[diff] [blame]111// CK2: call void @[[OFFL1:.+]]({{.+}})
112// CK2: {{%.+}} = call{{.*}} i32 @[[TMAIN:.+]]({{.+}})
113// CK2: ret
114
115// CK2: define {{.*}}void @[[OFFL1]]({{.+}})
116// CK2: call void {{.+}} @__kmpc_fork_teams({{.+}}, i32 5, {{.+}} @[[OUTL1:.+]] to {{.+}},
117// CK2: ret void
118
119// CK2: define internal void @[[OUTL1]]({{.+}})
120// CK2: [[OMP_UB:%.omp.ub]] = alloca i64,
121// CK2: store i64 {{.+}}, i64* [[OMP_UB]],
122// CK2: call void @__kmpc_for_static_init_8({{.+}}, {{.+}}, i32 92, {{.+}}, {{.+}}, i64* [[OMP_UB]],
123// CK2: call void @__kmpc_for_static_fini(
124// CK2: ret void
125// CK2: define {{.*}}i32 @[[TMAIN]]({{.+}})
Alexey Bataev50a1c782017-12-01 21:31:08 +0000[diff] [blame]126// CK2: call i32 @__tgt_target_teams(
Carlo Bertolliba1487b2017-10-04 14:12:09 +0000[diff] [blame]127// CK2: call void @[[OFFLT1:.+]]({{.+}})
128// CK2: ret
129// CK2-NEXT: }
130
131// CK2: define {{.*}}void @[[OFFLT1]]({{.+}})
132// CK2: call void {{.+}} @__kmpc_fork_teams({{.+}}, i32 1, {{.+}} @[[OUTLT1:.+]] to {{.+}},
133// CK2: ret void
134
135// CK2: define internal void @[[OUTLT1]]({{.+}})
136// discard loop variables not needed here
137// CK2: [[OMP_UB:%.omp.ub]] = alloca i32,
138// CK2: store i32 {{.+}}, i32* [[OMP_UB]],
139// CK2: call void @__kmpc_for_static_init_4({{.+}}, {{.+}}, i32 92, {{.+}}, {{.+}}, i32* [[OMP_UB]],
140// CK2: call void @__kmpc_for_static_fini(
141// CK2: ret void
142
143#endif // CK2
144#endif // #ifndef HEADER