124 lines · cpp
1// RUN: %clang_cc1 -triple x86_64-pc-linux-gnu -fopenmp \2// RUN: -fsyntax-only -verify %s3 4// RUN: %clang_cc1 -triple x86_64-pc-linux-gnu -fopenmp \5// RUN: -ast-print %s | FileCheck %s6 7// RUN: %clang_cc1 -triple x86_64-pc-linux-gnu -fopenmp \8// RUN: -emit-pch -o %t %s9 10// RUN: %clang_cc1 -triple x86_64-pc-linux-gnu -fopenmp \11// RUN: -include-pch %t -ast-print %s | FileCheck %s12 13// expected-no-diagnostics14 15#ifndef HEADER16#define HEADER17 18typedef void **omp_allocator_handle_t;19extern const omp_allocator_handle_t omp_null_allocator;20extern const omp_allocator_handle_t omp_default_mem_alloc;21extern const omp_allocator_handle_t omp_large_cap_mem_alloc;22extern const omp_allocator_handle_t omp_const_mem_alloc;23extern const omp_allocator_handle_t omp_high_bw_mem_alloc;24extern const omp_allocator_handle_t omp_low_lat_mem_alloc;25extern const omp_allocator_handle_t omp_cgroup_mem_alloc;26extern const omp_allocator_handle_t omp_pteam_mem_alloc;27extern const omp_allocator_handle_t omp_thread_mem_alloc;28 29//CHECK: template <typename T, int C> void templ_foo(T t) {30//CHECK: T j, z;31//CHECK: #pragma omp parallel loop collapse(C) reduction(+: z) lastprivate(j) bind(thread)32//CHECK: for (T i = 0; i < t; ++i)33//CHECK: for (j = 0; j < t; ++j)34//CHECK: z += i + j;35//CHECK: }36 37//CHECK: template<> void templ_foo<int, 2>(int t) {38//CHECK: int j, z;39//CHECK: #pragma omp parallel loop collapse(2) reduction(+: z) lastprivate(j) bind(thread)40//CHECK: for (int i = 0; i < t; ++i)41//CHECK: for (j = 0; j < t; ++j)42//CHECK: z += i + j;43//CHECK: }44template <typename T, int C>45void templ_foo(T t) {46 47 T j,z;48 #pragma omp parallel loop collapse(C) reduction(+:z) lastprivate(j) bind(thread)49 for (T i = 0; i<t; ++i)50 for (j = 0; j<t; ++j)51 z += i+j;52}53 54 55//CHECK: void test() {56void test() {57 constexpr int N = 100;58 float MTX[N][N];59 int aaa[1000];60 61 //CHECK: #pragma omp target teams distribute parallel for map(tofrom: MTX)62 //CHECK: #pragma omp parallel loop63 #pragma omp target teams distribute parallel for map(MTX)64 for (auto i = 0; i < N; ++i) {65 #pragma omp parallel loop66 for (auto j = 0; j < N; ++j) {67 MTX[i][j] = 0;68 }69 }70 71 //CHECK: #pragma omp target teams72 //CHECK: #pragma omp parallel loop73 #pragma omp target teams74 for (int i=0; i<1000; ++i) {75 #pragma omp parallel loop76 for (int j=0; j<100; j++) {77 aaa[i] += i + j;78 }79 }80 81 int j, z, z1, z2 = 1, z3 = 2;82 //CHECK: #pragma omp parallel loop collapse(2) private(z) lastprivate(j) order(concurrent) firstprivate(z2) num_threads(4) if(1) allocate(omp_default_mem_alloc: z2) shared(z3) reduction(+: z1) bind(parallel) proc_bind(primary)83 #pragma omp parallel loop collapse(2) private(z) lastprivate(j) \84 order(concurrent) firstprivate(z2) num_threads(4) if(1) \85 allocate(omp_default_mem_alloc:z2) shared(z3) \86 reduction(+:z1) bind(parallel) proc_bind(primary)87 for (auto i = 0; i < N; ++i) {88 for (j = 0; j < N; ++j) {89 z = i+j;90 MTX[i][j] = z;91 z1 += z;92 }93 }94 95 //CHECK: #pragma omp target teams96 //CHECK: #pragma omp parallel loop bind(teams)97 #pragma omp target teams98 #pragma omp parallel loop bind(teams)99 for (auto i = 0; i < N; ++i) { }100 101 //CHECK: #pragma omp target102 //CHECK: #pragma omp teams103 //CHECK: #pragma omp parallel loop bind(teams)104 #pragma omp target105 #pragma omp teams106 #pragma omp parallel loop bind(teams)107 for (auto i = 0; i < N; ++i) { }108}109 110//CHECK: void nobindingfunc() {111void nobindingfunc()112{113 //CHECK: #pragma omp parallel loop114 #pragma omp parallel loop115 for (int i=0; i<10; ++i) { }116}117 118void bar()119{120 templ_foo<int,2>(8);121}122 123#endif // HEADER124