Clang Project

clang_source_code/test/OpenMP/teams_distribute_parallel_for_num_threads_codegen.cpp
1// RUN: %clang_cc1 -verify -fopenmp -fopenmp-version=45 -fopenmp-targets=powerpc64le-ibm-linux-gnu -x c++ -triple powerpc64le-ibm-linux-gnu -emit-llvm %s -fexceptions -fcxx-exceptions -o - | FileCheck %s
2// RUN: %clang_cc1 -fopenmp -fopenmp-version=45 -fopenmp-targets=powerpc64le-ibm-linux-gnu -x c++ -std=c++11 -triple powerpc64le-ibm-linux-gnu -fexceptions -fcxx-exceptions -emit-pch -o %t %s
3// RUN: %clang_cc1 -fopenmp -fopenmp-version=45 -fopenmp-targets=powerpc64le-ibm-linux-gnu -x c++ -triple powerpc64le-ibm-linux-gnu -fexceptions -fcxx-exceptions -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck %s
4
5// RUN: %clang_cc1 -verify -fopenmp-simd -fopenmp-version=45 -fopenmp-targets=powerpc64le-ibm-linux-gnu -x c++ -triple powerpc64le-ibm-linux-gnu -emit-llvm %s -fexceptions -fcxx-exceptions -o - | FileCheck --check-prefix SIMD-ONLY0 %s
6// RUN: %clang_cc1 -fopenmp-simd -fopenmp-version=45 -fopenmp-targets=powerpc64le-ibm-linux-gnu -x c++ -std=c++11 -triple powerpc64le-ibm-linux-gnu -fexceptions -fcxx-exceptions -emit-pch -o %t %s
7// RUN: %clang_cc1 -fopenmp-simd -fopenmp-version=45 -fopenmp-targets=powerpc64le-ibm-linux-gnu -x c++ -triple powerpc64le-ibm-linux-gnu -fexceptions -fcxx-exceptions -std=c++11 -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#ifndef HEADER
11#define HEADER
12
13typedef __INTPTR_TYPE__ intptr_t;
14
15// CHECK-DAG: [[IDENT_T_TY:%.+]] = type { i32, i32, i32, i32, i8* }
16// CHECK-DAG: [[S_TY:%.+]] = type { [[INTPTR_T_TY:i[0-9]+]], [[INTPTR_T_TY]], [[INTPTR_T_TY]] }
17// CHECK-DAG: [[STR:@.+]] = private unnamed_addr constant [23 x i8] c";unknown;unknown;0;0;;\00"
18// CHECK-DAG: [[DEF_LOC_2:@.+]] = private unnamed_addr global [[IDENT_T_TY]] { i32 0, i32 2, i32 0, i32 0, i8* getelementptr inbounds ([23 x i8], [23 x i8]* [[STR]], i32 0, i32 0) }
19
20void foo();
21
22struct S {
23  intptr_t a, b, c;
24  S(intptr_t a) : a(a) {}
25  operator char() { return a; }
26  ~S() {}
27};
28
29template <typename T, int C>
30int tmain() {
31#pragma omp target
32#pragma omp teams distribute parallel for num_threads(C)
33  for (int i = 0; i < 100; i++)
34    foo();
35#pragma omp target
36#pragma omp teams distribute parallel for num_threads(T(23))
37  for (int i = 0; i < 100; i++)
38    foo();
39  return 0;
40}
41
42int main() {
43  S s(0);
44  char a = s;
45// CHECK: call i{{[0-9]+}} @__tgt_target_teams(
46// CHECK: call void [[OFFLOADING_FUN_0:@.+]](
47// CHECK: call i{{[0-9]+}} @__tgt_target_teams(
48// CHECK: call void [[OFFLOADING_FUN_1:@.+]](
49// CHECK: invoke{{.+}} [[TMAIN_5:@.+]]()
50// CHECK: invoke{{.+}} [[TMAIN_1:@.+]]()
51#pragma omp target
52  // CHECK: define internal void [[OFFLOADING_FUN_0]](
53  // CHECK: call {{.*}}void {{.+}} @__kmpc_fork_teams({{.+}}, i32 0, {{.+}}* [[OMP_TEAMS_OUTLINED_0:@.+]] to {{.+}})
54#pragma omp teams distribute parallel for num_threads(2)
55  for (int i = 0; i < 100; i++) {
56    // CHECK: define{{.+}} void [[OMP_TEAMS_OUTLINED_0]](
57    // CHECK:       call {{.*}}void @__kmpc_push_num_threads([[IDENT_T_TY]]* [[DEF_LOC_2]], i32 {{.+}}, i32 2)
58    // CHECK:       call {{.*}}void {{.*}} @__kmpc_fork_call(
59    foo();
60  }
61#pragma omp target
62  // CHECK: define internal void [[OFFLOADING_FUN_1]](
63
64  // CHECK: call {{.*}}void {{.+}} @__kmpc_fork_teams({{.+}}, i32 1, {{.+}}* [[OMP_TEAMS_OUTLINED_1:@.+]] to {{.+}})
65#pragma omp teams distribute parallel for num_threads(a)
66  for (int i = 0; i < 100; i++) {
67    // CHECK: define{{.+}} void [[OMP_TEAMS_OUTLINED_1]](
68    // CHECK-DAG: [[A_ADDR:%.+]] = alloca i64,
69    // CHECK-DAG: [[A_REF:%.+]] = bitcast i64* [[A_ADDR]] to i8*
70    // CHECK-DAG: [[A_VAL:%.+]] = load i8, i8* [[A_REF]],
71    // CHECK-DAG: [[A_EXT:%.+]] = sext i8 [[A_VAL]] to {{.+}}
72    // CHECK: call {{.*}}void @__kmpc_push_num_threads([[IDENT_T_TY]]* [[DEF_LOC_2]], i32 {{.+}}, i32 [[A_EXT]])
73    // CHECK: call {{.*}}void {{.*}} @__kmpc_fork_call(
74    foo();
75  }
76  return a + tmain<char, 5>() + tmain<S, 1>();
77}
78
79// tmain 5
80// CHECK-DAG: define {{.*}}i{{[0-9]+}} [[TMAIN_5]]()
81// CHECK: call i{{[0-9]+}} @__tgt_target_teams(
82// CHECK: call void [[T_OFFLOADING_FUN_0:@.+]](
83// CHECK: call i{{[0-9]+}} @__tgt_target_teams(
84// CHECK: call void [[T_OFFLOADING_FUN_1:@.+]](
85
86// tmain 1
87// CHECK-DAG: define {{.*}}i{{[0-9]+}} [[TMAIN_1]]()
88// CHECK: call i{{[0-9]+}} @__tgt_target_teams(
89// CHECK: call void [[T_OFFLOADING_FUN_2:@.+]](
90// CHECK: call i{{[0-9]+}} @__tgt_target_teams(
91// CHECK: call void [[T_OFFLOADING_FUN_3:@.+]](
92
93// CHECK: define internal void [[T_OFFLOADING_FUN_0]](
94// CHECK: call {{.*}}void {{.+}} @__kmpc_fork_teams({{.+}}, i32 0, {{.+}}* [[T_OMP_TEAMS_OUTLINED_0:@.+]] to {{.+}})
95
96// CHECK: define{{.+}} void [[T_OMP_TEAMS_OUTLINED_0]](
97// CHECK:       call {{.*}}void @__kmpc_push_num_threads([[IDENT_T_TY]]* [[DEF_LOC_2]], i32 {{.+}}, i32 5)
98// CHECK:       call {{.*}}void {{.*}} @__kmpc_fork_call(
99
100// CHECK: define internal void [[T_OFFLOADING_FUN_1]](
101// CHECK: call {{.*}}void {{.+}} @__kmpc_fork_teams({{.+}}, i32 0, {{.+}}* [[T_OMP_TEAMS_OUTLINED_1:@.+]] to {{.+}})
102
103// CHECK: define{{.+}} void [[T_OMP_TEAMS_OUTLINED_1]](
104// CHECK:       call {{.*}}void @__kmpc_push_num_threads([[IDENT_T_TY]]* [[DEF_LOC_2]], i32 {{.+}}, i32 23)
105// CHECK:       call {{.*}}void {{.*}} @__kmpc_fork_call(
106
107// CHECK: define internal void [[T_OFFLOADING_FUN_2]](
108// CHECK: call {{.*}}void {{.+}} @__kmpc_fork_teams({{.+}}, i32 0, {{.+}}* [[T_OMP_TEAMS_OUTLINED_2:@.+]] to {{.+}})
109
110// CHECK: define{{.+}} void [[T_OMP_TEAMS_OUTLINED_2]](
111// CHECK:       call {{.*}}void @__kmpc_push_num_threads([[IDENT_T_TY]]* [[DEF_LOC_2]], i32 {{.+}}, i32 1)
112// CHECK:       call {{.*}}void {{.*}} @__kmpc_fork_call(
113
114// CHECK: define internal void [[T_OFFLOADING_FUN_3]](
115// CHECK: call {{.*}}void {{.+}} @__kmpc_fork_teams({{.+}}, i32 {{.+}}, {{.+}}* [[T_OMP_TEAMS_OUTLINED_3:@.+]] to {{.+}})
116
117// CHECK: define{{.+}} void [[T_OMP_TEAMS_OUTLINED_3]]({{.+}}, {{.+}}, {{.+}} [[NUM_TH_CPT_IN:%.+]])
118// CHECK: [[NUM_TH_CPT:%.+]] = alloca i64,
119// CHECK: store {{.+}} [[NUM_TH_CPT_IN]], {{.+}} [[NUM_TH_CPT]],
120// CHECK: [[NUM_TH_REF:%.+]] = bitcast i64* [[NUM_TH_CPT]] to i8*
121// CHECK-DAG:   [[NUM_TH_VAL:%.+]] = load {{.+}}, {{.+}} [[NUM_TH_REF]],
122// CHECK-DAG:   [[NUM_TH_SEXT:%.+]] = sext i8 [[NUM_TH_VAL]] to {{.+}}
123// CHECK:       call {{.*}}void @__kmpc_push_num_threads([[IDENT_T_TY]]* [[DEF_LOC_2]], i32 {{.+}}, i32 [[NUM_TH_SEXT]])
124// CHECK:       call {{.*}}void {{.*}} @__kmpc_fork_call(
125#endif
126
CodeThinker, Enjoy your reading code