a22236120f
Implementation of the unroll directive introduced in OpenMP 5.1. Follows the approach from D76342 for the tile directive (i.e. AST-based, not using the OpenMPIRBuilder). Tries to use `llvm.loop.unroll.*` metadata where possible, but has to fall back to an AST representation of the outer loop if the partially unrolled generated loop is associated with another directive (because it needs to compute the number of iterations). Reviewed By: ABataev Differential Revision: https://reviews.llvm.org/D99459
128 lines
4.0 KiB
C++
128 lines
4.0 KiB
C++
// Check no warnings/errors
|
|
// RUN: %clang_cc1 -triple x86_64-pc-linux-gnu -fopenmp -fopenmp-version=51 -fsyntax-only -verify %s
|
|
// expected-no-diagnostics
|
|
|
|
// Check AST and unparsing
|
|
// RUN: %clang_cc1 -triple x86_64-pc-linux-gnu -fopenmp -fopenmp-version=51 -ast-dump %s | FileCheck %s --check-prefix=DUMP
|
|
// RUN: %clang_cc1 -triple x86_64-pc-linux-gnu -fopenmp -fopenmp-version=51 -ast-print %s | FileCheck %s --check-prefix=PRINT --match-full-lines
|
|
|
|
// Check same results after serialization round-trip
|
|
// RUN: %clang_cc1 -triple x86_64-pc-linux-gnu -fopenmp -fopenmp-version=51 -emit-pch -o %t %s
|
|
// RUN: %clang_cc1 -triple x86_64-pc-linux-gnu -fopenmp -fopenmp-version=51 -include-pch %t -ast-dump-all %s | FileCheck %s --check-prefix=DUMP
|
|
// RUN: %clang_cc1 -triple x86_64-pc-linux-gnu -fopenmp -fopenmp-version=51 -include-pch %t -ast-print %s | FileCheck %s --check-prefix=PRINT --match-full-lines
|
|
|
|
#ifndef HEADER
|
|
#define HEADER
|
|
|
|
// placeholder for loop body code.
|
|
void body(...);
|
|
|
|
|
|
// PRINT-LABEL: void func_unroll() {
|
|
// DUMP-LABEL: FunctionDecl {{.*}} func_unroll
|
|
void func_unroll() {
|
|
// PRINT: #pragma omp unroll
|
|
// DUMP: OMPUnrollDirective
|
|
#pragma omp unroll
|
|
// PRINT-NEXT: for (int i = 7; i < 17; i += 3)
|
|
// DUMP-NEXT: ForStmt
|
|
for (int i = 7; i < 17; i += 3)
|
|
// PRINT-NEXT: body(i);
|
|
// DUMP: CallExpr
|
|
body(i);
|
|
}
|
|
|
|
|
|
// PRINT-LABEL: void func_unroll_full() {
|
|
// DUMP-LABEL: FunctionDecl {{.*}} func_unroll_full
|
|
void func_unroll_full() {
|
|
// PRINT: #pragma omp unroll full
|
|
// DUMP: OMPUnrollDirective
|
|
// DUMP-NEXT: OMPFullClause
|
|
#pragma omp unroll full
|
|
// PRINT-NEXT: for (int i = 7; i < 17; i += 3)
|
|
// DUMP-NEXT: ForStmt
|
|
for (int i = 7; i < 17; i += 3)
|
|
// PRINT-NEXT: body(i);
|
|
// DUMP: CallExpr
|
|
body(i);
|
|
}
|
|
|
|
|
|
// PRINT-LABEL: void func_unroll_partial() {
|
|
// DUMP-LABEL: FunctionDecl {{.*}} func_unroll_partial
|
|
void func_unroll_partial() {
|
|
// PRINT: #pragma omp unroll partial
|
|
// DUMP: OMPUnrollDirective
|
|
// DUMP-NEXT: OMPPartialClause
|
|
// DUMP-NEXT: <<<NULL>>>
|
|
#pragma omp unroll partial
|
|
// PRINT-NEXT: for (int i = 7; i < 17; i += 3)
|
|
// DUMP-NEXT: ForStmt
|
|
for (int i = 7; i < 17; i += 3)
|
|
// PRINT: body(i);
|
|
// DUMP: CallExpr
|
|
body(i);
|
|
}
|
|
|
|
|
|
// PRINT-LABEL: void func_unroll_partial_factor() {
|
|
// DUMP-LABEL: FunctionDecl {{.*}} func_unroll_partial_factor
|
|
void func_unroll_partial_factor() {
|
|
// PRINT: #pragma omp unroll partial(4)
|
|
// DUMP: OMPUnrollDirective
|
|
// DUMP-NEXT: OMPPartialClause
|
|
// DUMP-NEXT: ConstantExpr
|
|
// DUMP-NEXT: value: Int 4
|
|
// DUMP-NEXT: IntegerLiteral {{.*}} 4
|
|
#pragma omp unroll partial(4)
|
|
// PRINT-NEXT: for (int i = 7; i < 17; i += 3)
|
|
// DUMP-NEXT: ForStmt
|
|
for (int i = 7; i < 17; i += 3)
|
|
// PRINT-NEXT: body(i);
|
|
// DUMP: CallExpr
|
|
body(i);
|
|
}
|
|
|
|
|
|
// PRINT-LABEL: void func_unroll_partial_factor_for() {
|
|
// DUMP-LABEL: FunctionDecl {{.*}} func_unroll_partial_factor_for
|
|
void func_unroll_partial_factor_for() {
|
|
// PRINT: #pragma omp for
|
|
// DUMP: OMPForDirective
|
|
#pragma omp for
|
|
// PRINT: #pragma omp unroll partial(2)
|
|
// DUMP: OMPUnrollDirective
|
|
// DUMP-NEXT: OMPPartialClause
|
|
#pragma omp unroll partial(2)
|
|
// PRINT-NEXT: for (int i = 7; i < 17; i += 3)
|
|
// DUMP: ForStmt
|
|
for (int i = 7; i < 17; i += 3)
|
|
// PRINT-NEXT: body(i);
|
|
// DUMP: CallExpr
|
|
body(i);
|
|
}
|
|
|
|
|
|
// PRINT-LABEL: template <typename T, T Start, T End, T Step, int Factor> void unroll_templated() {
|
|
// DUMP-LABEL: FunctionTemplateDecl {{.*}} unroll_templated
|
|
template<typename T, T Start, T End, T Step, int Factor>
|
|
void unroll_templated() {
|
|
// PRINT: #pragma omp unroll partial(Factor)
|
|
// DUMP: OMPUnrollDirective
|
|
// DUMP-NEXT: OMPPartialClause
|
|
// DUMP-NEXT: DeclRefExpr {{.*}} 'Factor' 'int'
|
|
#pragma omp unroll partial(Factor)
|
|
// PRINT-NEXT: for (T i = Start; i < End; i += Step)
|
|
// DUMP-NEXT: ForStmt
|
|
for (T i = Start; i < End; i += Step)
|
|
// PRINT-NEXT: body(i);
|
|
// DUMP: CallExpr
|
|
body(i);
|
|
}
|
|
void unroll_template() {
|
|
unroll_templated<int,0,1024,1,4>();
|
|
}
|
|
|
|
#endif
|