Add parsing, sema analysis and serialization/deserialization for 'declare reduction' construct. User-defined reductions are defined as #pragma omp declare reduction( reduction-identifier : typename-list : combiner ) [initializer ( initializer-expr )] These custom reductions may be used in 'reduction' clauses of OpenMP constructs. The combiner specifies how partial results can be combined into a single value. The combiner can use the special variable identifiers omp_in and omp_out that are of the type of the variables being reduced with this reduction-identifier. Each of them will denote one of the values to be combined before executing the combiner. It is assumed that the special omp_out identifier will refer to the storage that holds the resulting combined value after executing the combiner. As the initializer-expr value of a user-defined reduction is not known a priori the initializer-clause can be used to specify one. Then the contents of the initializer-clause will be used as the initializer for private copies of reduction list items where the omp_priv identifier will refer to the storage to be initialized. The special identifier omp_orig can also appear in the initializer-clause and it will refer to the storage of the original variable to be reduced. Differential Revision: http://reviews.llvm.org/D11182 llvm-svn: 262582
70 lines
2.4 KiB
C++
70 lines
2.4 KiB
C++
// RUN: %clang_cc1 -verify -fopenmp -ast-print %s | FileCheck %s
|
|
// RUN: %clang_cc1 -fopenmp -x c++ -std=c++11 -emit-pch -o %t %s
|
|
// RUN: %clang_cc1 -fopenmp -std=c++11 -include-pch %t -fsyntax-only -verify %s -ast-print | FileCheck %s
|
|
// expected-no-diagnostics
|
|
|
|
#ifndef HEADER
|
|
#define HEADER
|
|
|
|
#pragma omp declare reduction(+ : int, char : omp_out *= omp_in)
|
|
// CHECK: #pragma omp declare reduction (+ : int : omp_out *= omp_in)
|
|
// CHECK-NEXT: #pragma omp declare reduction (+ : char : omp_out *= omp_in)
|
|
|
|
// CHECK: #pragma omp declare reduction (fun : int : omp_out += omp_in) initializer(omp_priv = omp_orig + 15)
|
|
|
|
template <class T>
|
|
class SSS {
|
|
public:
|
|
#pragma omp declare reduction(fun : T : omp_out += omp_in) initializer(omp_priv = omp_orig + 15)
|
|
// CHECK: #pragma omp declare reduction (fun : T : omp_out += omp_in) initializer(omp_priv = omp_orig + 15)
|
|
};
|
|
|
|
SSS<int> d;
|
|
|
|
void init(SSS<int> &lhs, SSS<int> rhs);
|
|
|
|
#pragma omp declare reduction(fun : SSS < int > : omp_out = omp_in) initializer(init(omp_priv, omp_orig))
|
|
// CHECK: #pragma omp declare reduction (fun : SSS<int> : omp_out = omp_in) initializer(init(omp_priv, omp_orig))
|
|
|
|
// CHECK: template <typename T = int> int foo(int a) {
|
|
// CHECK: #pragma omp declare reduction (fun : int : omp_out += omp_in) initializer(omp_priv = omp_orig + 15);
|
|
// CHECK: {
|
|
// CHECK: #pragma omp declare reduction (fun : int : omp_out += omp_in) initializer(omp_priv = omp_orig + 15);
|
|
// CHECK: }
|
|
// CHECK: return a;
|
|
// CHECK: }
|
|
|
|
// CHECK: template <typename T> T foo(T a) {
|
|
// CHECK: #pragma omp declare reduction (fun : T : omp_out += omp_in) initializer(omp_priv = omp_orig + 15);
|
|
// CHECK: {
|
|
// CHECK: #pragma omp declare reduction (fun : T : omp_out += omp_in) initializer(omp_priv = omp_orig + 15);
|
|
// CHECK: }
|
|
// CHECK: return a;
|
|
// CHECK: }
|
|
template <typename T>
|
|
T foo(T a) {
|
|
#pragma omp declare reduction(fun : T : omp_out += omp_in) initializer(omp_priv = omp_orig + 15)
|
|
{
|
|
#pragma omp declare reduction(fun : T : omp_out += omp_in) initializer(omp_priv = omp_orig + 15)
|
|
}
|
|
return a;
|
|
}
|
|
|
|
int main() {
|
|
int i = 0;
|
|
SSS<int> sss;
|
|
// TODO: Add support for scoped reduction identifiers
|
|
// #pragma omp parallel reduction(SSS<int>::fun : i)
|
|
// TODO-CHECK: #pragma omp parallel reduction(SSS<int>::fun: i)
|
|
{
|
|
i += 1;
|
|
}
|
|
// #pragma omp parallel reduction(::fun:sss)
|
|
// TODO-CHECK: #pragma omp parallel reduction(::fun: sss)
|
|
{
|
|
}
|
|
return foo(15);
|
|
}
|
|
|
|
#endif
|