73 lines · cpp
1// RUN: rm -rf %t2// RUN: split-file %s %t3// RUN: cd %t4 5// RUN: %clang_cc1 -std=c++20 -fopenmp -triple x86_64-unknown-linux-gnu %t/AssumeMod.cppm -emit-module-interface -o %t/AssumeMod.pcm6// 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.cppm7 8// expected-no-diagnostics9 10//--- AssumeMod.cppm11module;12export module AssumeMod;13export template<int N> int foo(int y) {14 int x = -1;15#pragma omp assume holds(y == N)16 {17 x = y;18 }19 return x;20}21 22export template<bool B> void bar(int *z) {23 if constexpr (B) {24#pragma omp assume no_openmp25 {26 z[0]++;27 }28 } else {29#pragma omp assume contains(target, parallel) absent(loop)30 {31 z[1] += 2;32 }33 }34}35 36// CHECK: FunctionTemplateDecl 0x{{.*}}37// CHECK: OMPAssumeDirective 0x{{.*}} <line:5:1, col:33>38// CHECK-NEXT: OMPHoldsClause 0x{{.*}} <col:20, col:32>39 40// CHECK: FunctionDecl 0x{{.*}} implicit_instantiation41// CHECK: OMPAssumeDirective 0x{{.*}} <line:5:1, col:33>42// CHECK-NEXT: OMPHoldsClause 0x{{.*}} <col:20, col:32>43 44// CHECK: FunctionTemplateDecl 0x{{.*}}45// CHECK: OMPAssumeDirective 0x{{.*}} <line:14:1, col:29>46// CHECK-NEXT: OMPNo_openmpClause 0x{{.*}} <col:20, col:29>47// CHECK: OMPAssumeDirective 0x{{.*}} <line:19:1, col:59>48// CHECK-NEXT: OMPContainsClause 0x{{.*}} <col:20, col:45>49// CHECK-NEXT: OMPAbsentClause 0x{{.*}} <col:47, col:58>50 51// CHECK: FunctionDecl 0x{{.*}} implicit_instantiation52// CHECK: OMPAssumeDirective 0x{{.*}} <line:14:1, col:29>53// CHECK-NEXT: OMPNo_openmpClause 0x{{.*}} <col:20, col:29>54 55// CHECK: FunctionDecl 0x{{.*}} implicit_instantiation56// CHECK: OMPAssumeDirective 0x{{.*}} <line:19:1, col:59>57// CHECK-NEXT: OMPContainsClause 0x{{.*}} <col:20, col:45>58// CHECK-NEXT: OMPAbsentClause 0x{{.*}} <col:47, col:58>59 60//--- UseAssumeMod.cpp61import AssumeMod;62 63extern "C" int printf(const char* fmt, ...);64 65int main() {66 printf ("foo(5)=%d\n", foo<5> (5));67 int arr[2] = { 0, 0 };68 bar<true>(arr);69 bar<false>(arr);70 printf ("arr[0]=%d, arr[1]=%d\n", arr[0], arr[1]);71 return 0;72}73