llvm-project/clang/test/OpenMP/assume_serialize_deserialize_tmpl.cpp
Julian Brown a42e515e3a
[OpenMP] OpenMP 5.1 "assume" directive parsing support (#92731)
This is a minimal patch to support parsing for "omp assume" directives.
These are meant to be hints to a compiler's optimisers: as such, it is
legitimate (if not very useful) to ignore them. The patch builds on top
of the existing support for "omp assumes" directives (note spelling!).

Unlike the "omp [begin/end] assumes" directives, "omp assume" is
associated with a compound statement, i.e. it can appear within a
function. The "holds" assumption could (theoretically) be mapped onto
the existing builtin "__builtin_assume", though the latter applies to a
single point in the program, and the former to a range (i.e. the whole
of the associated compound statement).

This patch fixes sollve's OpenMP 5.1 "omp assume"-based tests.
2024-08-05 07:37:07 -04:00

73 lines
2.1 KiB
C++

// RUN: rm -rf %t
// RUN: split-file %s %t
// RUN: cd %t
// RUN: %clang_cc1 -std=c++20 -fopenmp -triple x86_64-unknown-linux-gnu %t/AssumeMod.cppm -emit-module-interface -o %t/AssumeMod.pcm
// RUN: %clang_cc1 -std=c++20 -fopenmp -triple x86_64-unknown-linux-gnu %t/UseAssumeMod.cpp -fmodule-file=AssumeMod=%t/AssumeMod.pcm -ast-dump-all | FileCheck %t/AssumeMod.cppm
// expected-no-diagnostics
//--- AssumeMod.cppm
module;
export module AssumeMod;
export template<int N> int foo(int y) {
int x = -1;
#pragma omp assume holds(y == N)
{
x = y;
}
return x;
}
export template<bool B> void bar(int *z) {
if constexpr (B) {
#pragma omp assume no_openmp
{
z[0]++;
}
} else {
#pragma omp assume contains(target, parallel) absent(loop)
{
z[1] += 2;
}
}
}
// CHECK: FunctionTemplateDecl 0x{{.*}}
// CHECK: OMPAssumeDirective 0x{{.*}} <line:5:1, col:33>
// CHECK-NEXT: OMPHoldsClause 0x{{.*}} <col:20, col:32>
// CHECK: FunctionDecl 0x{{.*}} implicit_instantiation
// CHECK: OMPAssumeDirective 0x{{.*}} <line:5:1, col:33>
// CHECK-NEXT: OMPHoldsClause 0x{{.*}} <col:20, col:32>
// CHECK: FunctionTemplateDecl 0x{{.*}}
// CHECK: OMPAssumeDirective 0x{{.*}} <line:14:1, col:29>
// CHECK-NEXT: OMPNo_openmpClause 0x{{.*}} <col:20, col:29>
// CHECK: OMPAssumeDirective 0x{{.*}} <line:19:1, col:59>
// CHECK-NEXT: OMPContainsClause 0x{{.*}} <col:20, col:45>
// CHECK-NEXT: OMPAbsentClause 0x{{.*}} <col:47, col:58>
// CHECK: FunctionDecl 0x{{.*}} implicit_instantiation
// CHECK: OMPAssumeDirective 0x{{.*}} <line:14:1, col:29>
// CHECK-NEXT: OMPNo_openmpClause 0x{{.*}} <col:20, col:29>
// CHECK: FunctionDecl 0x{{.*}} implicit_instantiation
// CHECK: OMPAssumeDirective 0x{{.*}} <line:19:1, col:59>
// CHECK-NEXT: OMPContainsClause 0x{{.*}} <col:20, col:45>
// CHECK-NEXT: OMPAbsentClause 0x{{.*}} <col:47, col:58>
//--- UseAssumeMod.cpp
import AssumeMod;
extern "C" int printf(const char* fmt, ...);
int main() {
printf ("foo(5)=%d\n", foo<5> (5));
int arr[2] = { 0, 0 };
bar<true>(arr);
bar<false>(arr);
printf ("arr[0]=%d, arr[1]=%d\n", arr[0], arr[1]);
return 0;
}