befb4be3a8
This is the second part loosely extracted from D71179 and cleaned up.
This patch provides semantic analysis support for `omp begin/end declare
variant`, mostly as defined in OpenMP technical report 8 (TR8) [0].
The sema handling makes code generation obsolete as we generate "the
right" calls that can just be handled as usual. This handling also
applies to the existing, albeit problematic, `omp declare variant
support`. As a consequence a lot of unneeded code generation and
complexity is removed.
A major purpose of this patch is to provide proper `math.h`/`cmath`
support for OpenMP target offloading. See PR42061, PR42798, PR42799. The
current code was developed with this feature in mind, see [1].
The logic is as follows:
If we have seen a `#pragma omp begin declare variant match(<SELECTOR>)`
but not the corresponding `end declare variant`, and we find a function
definition we will:
1) Create a function declaration for the definition we were about to generate.
2) Create a function definition but with a mangled name (according to
`<SELECTOR>`).
3) Annotate the declaration with the `OMPDeclareVariantAttr`, the same
one used already for `omp declare variant`, using and the mangled
function definition as specialization for the context defined by
`<SELECTOR>`.
When a call is created we inspect it. If the target has an
`OMPDeclareVariantAttr` attribute we try to specialize the call. To this
end, all variants are checked, the best applicable one is picked and a
new call to the specialization is created. The new call is used instead
of the original one to the base function. To keep the AST printing and
tooling possible we utilize the PseudoObjectExpr. The original call is
the syntactic expression, the specialized call is the semantic
expression.
[0] https://www.openmp.org/wp-content/uploads/openmp-TR8.pdf
[1] https://reviews.llvm.org/D61399#change-496lQkg0mhRN
Reviewers: kiranchandramohan, ABataev, RaviNarayanaswamy, gtbercea, grokos, sdmitriev, JonChesterfield, hfinkel, fghanim, aaron.ballman
Subscribers: bollu, guansong, openmp-commits, cfe-commits
Tags: #clang
Differential Revision: https://reviews.llvm.org/D75779
155 lines
5.5 KiB
C++
155 lines
5.5 KiB
C++
// 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 -fopenmp-version=50 -DGPU
|
|
// RUN: %clang_cc1 -verify -fopenmp -x c++ -triple nvptx64-unknown-unknown -aux-triple powerpc64le-unknown-unknown -emit-llvm %s -fopenmp-is-device -fopenmp-host-ir-file-path %t-ppc-host.bc -o - -fopenmp-version=50 -DGPU | FileCheck %s --implicit-check-not='ret i32 {{6|7|9|10|12|14|17|18|20|21|22|23|24|26}}'
|
|
// RUN: %clang_cc1 -verify -fopenmp -x c++ -triple nvptx64-unknown-unknown -aux-triple powerpc64le-unknown-unknown -emit-llvm %s -fopenmp-is-device -fopenmp-host-ir-file-path %t-ppc-host.bc -emit-pch -o %t -fopenmp-version=50 -DGPU
|
|
// RUN: %clang_cc1 -verify -fopenmp -x c++ -triple nvptx64-unknown-unknown -aux-triple powerpc64le-unknown-unknown -emit-llvm %s -fopenmp-is-device -fopenmp-host-ir-file-path %t-ppc-host.bc -include-pch %t -o - -fopenmp-version=50 -DGPU | FileCheck %s --implicit-check-not='ret i32 {{6|7|9|10|12|14|17|18|20|21|22|23|24|26}}'
|
|
|
|
// 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 -fopenmp-version=50 -DNOHOST
|
|
// RUN: %clang_cc1 -verify -fopenmp -x c++ -triple nvptx64-unknown-unknown -aux-triple powerpc64le-unknown-unknown -emit-llvm %s -fopenmp-is-device -fopenmp-host-ir-file-path %t-ppc-host.bc -o - -fopenmp-version=50 -DNOHOST | FileCheck %s --implicit-check-not='ret i32 {{6|7|9|10|12|14|17|18|20|21|22|23|24|26}}'
|
|
// RUN: %clang_cc1 -verify -fopenmp -x c++ -triple nvptx64-unknown-unknown -aux-triple powerpc64le-unknown-unknown -emit-llvm %s -fopenmp-is-device -fopenmp-host-ir-file-path %t-ppc-host.bc -emit-pch -o %t -fopenmp-version=50 -DNOHOST
|
|
// RUN: %clang_cc1 -verify -fopenmp -x c++ -triple nvptx64-unknown-unknown -aux-triple powerpc64le-unknown-unknown -emit-llvm %s -fopenmp-is-device -fopenmp-host-ir-file-path %t-ppc-host.bc -include-pch %t -o - -fopenmp-version=50 -DNOHOST | FileCheck %s --implicit-check-not='ret i32 {{6|7|9|10|12|14|17|18|20|21|22|23|24|26}}'
|
|
// expected-no-diagnostics
|
|
|
|
// CHECK-DAG: ret i32 2
|
|
// CHECK-DAG: ret i32 3
|
|
// CHECK-DAG: ret i32 4
|
|
// CHECK-DAG: ret i32 5
|
|
// CHECK-DAG: ret i32 8
|
|
// CHECK-DAG: ret i32 11
|
|
// CHECK-DAG: ret i32 13
|
|
// CHECK-DAG: ret i32 15
|
|
// CHECK-DAG: ret i32 16
|
|
// CHECK-DAG: ret i32 19
|
|
// CHECK-DAG: ret i32 25
|
|
|
|
// Outputs for function members checked via implicit filecheck flag
|
|
|
|
|
|
#ifndef HEADER
|
|
#define HEADER
|
|
|
|
#ifdef GPU
|
|
#define SUBSET gpu
|
|
#define CORRECT nohost, gpu
|
|
#define WRONG cpu, gpu
|
|
#endif // GPU
|
|
#ifdef NOHOST
|
|
#define SUBSET nohost
|
|
#define CORRECT nohost, gpu
|
|
#define WRONG nohost, host
|
|
#endif // NOHOST
|
|
|
|
int foo() { return 2; }
|
|
int bazzz();
|
|
int test();
|
|
static int stat_unused_();
|
|
static int stat_used_();
|
|
|
|
#pragma omp declare target
|
|
|
|
#pragma omp declare variant(foo) match(device = {kind(CORRECT)})
|
|
int bar() { return 3; }
|
|
|
|
#pragma omp declare variant(bazzz) match(device = {kind(CORRECT)})
|
|
int baz() { return 4; }
|
|
|
|
#pragma omp declare variant(test) match(device = {kind(CORRECT)})
|
|
int call() { return 5; }
|
|
|
|
#pragma omp declare variant(stat_unused_) match(device = {kind(CORRECT)})
|
|
static int stat_unused() { return 6; }
|
|
|
|
#pragma omp declare variant(stat_used_) match(device = {kind(CORRECT)})
|
|
static int stat_used() { return 7; }
|
|
|
|
#pragma omp end declare target
|
|
|
|
int main() {
|
|
int res;
|
|
#pragma omp target map(from \
|
|
: res)
|
|
res = bar() + baz() + call();
|
|
return res;
|
|
}
|
|
|
|
int test() { return 8; }
|
|
static int stat_unused_() { return 9; }
|
|
static int stat_used_() { return 10; }
|
|
|
|
#pragma omp declare target
|
|
|
|
struct SpecialFuncs {
|
|
void vd() {}
|
|
SpecialFuncs();
|
|
~SpecialFuncs();
|
|
int method_() { return 11; }
|
|
#pragma omp declare variant(SpecialFuncs::method_) \
|
|
match(device = {kind(CORRECT)})
|
|
int method() { return 12; }
|
|
#pragma omp declare variant(SpecialFuncs::method_) \
|
|
match(device = {kind(CORRECT)})
|
|
int Method();
|
|
} s;
|
|
|
|
int SpecialFuncs::Method() { return 13; }
|
|
|
|
struct SpecSpecialFuncs {
|
|
void vd() {}
|
|
SpecSpecialFuncs();
|
|
~SpecSpecialFuncs();
|
|
|
|
int method_();
|
|
#pragma omp declare variant(SpecSpecialFuncs::method_) \
|
|
match(device = {kind(CORRECT)})
|
|
int method() { return 14; }
|
|
#pragma omp declare variant(SpecSpecialFuncs::method_) \
|
|
match(device = {kind(CORRECT)})
|
|
int Method();
|
|
} s1;
|
|
|
|
#pragma omp end declare target
|
|
|
|
int SpecSpecialFuncs::method_() { return 15; }
|
|
int SpecSpecialFuncs::Method() { return 16; }
|
|
|
|
int prio() { return 17; }
|
|
int prio1() { return 18; }
|
|
static int prio2() { return 19; }
|
|
static int prio3() { return 20; }
|
|
static int prio4() { return 21; }
|
|
int fn_linkage_variant() { return 22; }
|
|
extern "C" int fn_linkage_variant1() { return 23; }
|
|
int fn_variant2() { return 24; }
|
|
|
|
#pragma omp declare target
|
|
|
|
void xxx() {
|
|
(void)s.method();
|
|
(void)s1.method();
|
|
}
|
|
|
|
#pragma omp declare variant(prio) match(device = {kind(SUBSET)})
|
|
#pragma omp declare variant(prio1) match(device = {kind(CORRECT)})
|
|
int prio_() { return 25; }
|
|
|
|
#pragma omp declare variant(prio4) match(device = {kind(SUBSET)})
|
|
#pragma omp declare variant(prio2) match(device = {kind(CORRECT)})
|
|
#pragma omp declare variant(prio3) match(device = {kind(SUBSET)})
|
|
static int prio1_() { return 26; }
|
|
|
|
int int_fn() { return prio1_(); }
|
|
|
|
extern "C" {
|
|
#pragma omp declare variant(fn_linkage_variant) match(device = {kind(CORRECT)})
|
|
int fn_linkage() { return 27; }
|
|
}
|
|
|
|
#pragma omp declare variant(fn_linkage_variant1) match(device = {kind(CORRECT)})
|
|
int fn_linkage1() { return 28; }
|
|
|
|
#pragma omp declare variant(fn_variant2) match(device = {kind(WRONG)})
|
|
int fn2() { return 29; }
|
|
|
|
#pragma omp end declare target
|
|
|
|
#endif // HEADER
|