Peano
Loading...
Searching...
No Matches
Loop.h
Go to the documentation of this file.
1// This file is part of the Peano project. For conditions of distribution and
2// use, please see the copyright notice at www.peano-framework.org
3//
4// Included by peano4/utils/Loop.h. Do not use directly
5//
6
7// TODO: It does not work if you have multiple pragmas with if, so I only support taskloop
8// as I assume that parallel for in nested parallelism would not work anyway.
9// TODO: Use variadic templates for n-dimensions. -> Might not work, but maybe there is some (similar) trick.
10
11#define parallelForWithSchedulerInstructions(counter, max, loopParallelism) \
12 { \
13 auto __loopParallelismUsedWithinMacroWithUniqueName = loopParallelism; \
14 _Pragma( "omp taskloop default(shared) if ( __loopParallelismUsedWithinMacroWithUniqueName==::peano4::utils::LoopPlacement::SpreadOut)" ) \
15 for( int counter = 0; counter < max; counter++ ) {
16
17#define endParallelFor }}
18
19
20#define parallelDforWithSchedulerInstructions2d(counter, max, loopParallelism) \
21 { \
22 tarch::la::Vector<2,int> counter##Max(max); \
23 auto __loopParallelismUsedWithinMacroWithUniqueName = loopParallelism; \
24 _Pragma( "omp taskloop collapse(2) default(shared) if (__loopParallelismUsedWithinMacroWithUniqueName==::peano4::utils::LoopPlacement::SpreadOut)" ) \
25 for( int counter##0 = 0; counter##0 < counter##Max(0); counter##0++ ) \
26 for( int counter##1 = 0; counter##1 < counter##Max(1); counter##1++ ) { \
27 tarch::la::Vector<2, int> counter = { counter##0, counter##1 };
28
29#define parallelDforWithSchedulerInstructions3d(counter, max, loopParallelism) \
30 { \
31 tarch::la::Vector<3,int> counter##Max(max); \
32 auto __loopParallelismUsedWithinMacroWithUniqueName = loopParallelism; \
33 _Pragma( "omp taskloop collapse(3) default(shared) if (__loopParallelismUsedWithinMacroWithUniqueName==::peano4::utils::LoopPlacement::SpreadOut)" ) \
34 for( int counter##0 = 0; counter##0 < counter##Max(0); counter##0++ ) \
35 for( int counter##1 = 0; counter##1 < counter##Max(1); counter##1++ ) \
36 for( int counter##2 = 0; counter##2 < counter##Max(2); counter##2++ ) { \
37 tarch::la::Vector<3, int> counter = { counter##0, counter##1, counter##2 };
38
39#define parallelDforWithSchedulerInstructions4d(counter, max, loopParallelism) \
40 { \
41 tarch::la::Vector<4,int> counter##Max(max); \
42 auto __loopParallelismUsedWithinMacroWithUniqueName = loopParallelism; \
43 _Pragma( "omp taskloop collapse(4) default(shared) if (__loopParallelismUsedWithinMacroWithUniqueName==::peano4::utils::LoopPlacement::SpreadOut)" ) \
44 for( int counter##0 = 0; counter##0 < counter##Max(0); counter##0++ ) \
45 for( int counter##1 = 0; counter##1 < counter##Max(1); counter##1++ ) \
46 for( int counter##2 = 0; counter##2 < counter##Max(2); counter##2++ ) \
47 for( int counter##3 = 0; counter##3 < counter##Max(3); counter##3++ ) { \
48 tarch::la::Vector<4, int> counter = { counter##0, counter##1, counter##2, counter##3 };
49
50#define parallelDforWithSchedulerInstructions5d(counter, max, loopParallelism) \
51 { \
52 tarch::la::Vector<5,int> counter##Max(max); \
53 auto __loopParallelismUsedWithinMacroWithUniqueName = loopParallelism; \
54 _Pragma( "omp taskloop collapse(5) default(shared) if (__loopParallelismUsedWithinMacroWithUniqueName==::peano4::utils::LoopPlacement::SpreadOut)" ) \
55 for( int counter##0 = 0; counter##0 < counter##Max(0); counter##0++ ) \
56 for( int counter##1 = 0; counter##1 < counter##Max(1); counter##1++ ) \
57 for( int counter##2 = 0; counter##2 < counter##Max(2); counter##2++ ) \
58 for( int counter##3 = 0; counter##3 < counter##Max(3); counter##3++ ) \
59 for( int counter##4 = 0; counter##4 < counter##Max(4); counter##4++ ) { \
60 tarch::la::Vector<5, int> counter = { counter##0, counter##1, counter##2, counter##3, counter##4 };
61
62#define endParallelDfor }}
63
64
65#define simtForWithSchedulerInstructions(counter, max, loopParallelism) \
66 { \
67 auto __loopParallelismUsedWithinMacroWithUniqueName = loopParallelism; \
68 _Pragma( "omp taskloop simd default(shared) if ( __loopParallelismUsedWithinMacroWithUniqueName!=::peano4::utils::LoopPlacement::Serial)" ) \
69 for( int counter = 0; counter < max; counter++ ) {
70
71#define endSimtFor }}
72
73
113#define simtDforWithSchedulerInstructions2d(counter, max, loopParallelism) \
114 { \
115 tarch::la::Vector<2,int> counter##Max(max); \
116 auto __loopParallelismUsedWithinMacroWithUniqueName = loopParallelism; \
117 _Pragma( "omp parallel for if (__loopParallelismUsedWithinMacroWithUniqueName==::peano4::utils::LoopPlacement::SpreadOut)" ) \
118 for( int counter##0 = 0; counter##0 < counter##Max(0); counter##0++ ) { \
119 _Pragma( "omp simd if (__loopParallelismUsedWithinMacroWithUniqueName!=::peano4::utils::LoopPlacement::Serial)" ) \
120 for( int counter##1 = 0; counter##1 < counter##Max(1); counter##1++ ) { \
121 tarch::la::Vector<2, int> counter = { counter##0, counter##1 };
122
123#define simtDforWithSchedulerInstructions3d(counter, max, loopParallelism) \
124 { \
125 tarch::la::Vector<3,int> counter##Max(max); \
126 auto __loopParallelismUsedWithinMacroWithUniqueName = loopParallelism; \
127 _Pragma( "omp parallel for if (__loopParallelismUsedWithinMacroWithUniqueName==::peano4::utils::LoopPlacement::SpreadOut)" ) \
128 for( int counter##0 = 0; counter##0 < counter##Max(0); counter##0++ ) \
129 for( int counter##1 = 0; counter##1 < counter##Max(1); counter##1++ ) { \
130 _Pragma( "omp simd if (__loopParallelismUsedWithinMacroWithUniqueName!=::peano4::utils::LoopPlacement::Serial)" ) \
131 for( int counter##2 = 0; counter##2 < counter##Max(2); counter##2++ ) { \
132 tarch::la::Vector<3, int> counter = { counter##0, counter##1, counter##2 };
133
134#define simtDforWithSchedulerInstructions4d(counter, max, loopParallelism) \
135 { \
136 tarch::la::Vector<4,int> counter##Max(max); \
137 auto __loopParallelismUsedWithinMacroWithUniqueName = loopParallelism; \
138 _Pragma( "omp parallel for if (__loopParallelismUsedWithinMacroWithUniqueName==::peano4::utils::LoopPlacement::SpreadOut)" ) \
139 for( int counter##0 = 0; counter##0 < counter##Max(0); counter##0++ ) \
140 for( int counter##1 = 0; counter##1 < counter##Max(1); counter##1++ ) \
141 for( int counter##2 = 0; counter##2 < counter##Max(2); counter##2++ ) { \
142 _Pragma( "omp simd if (__loopParallelismUsedWithinMacroWithUniqueName!=::peano4::utils::LoopPlacement::Serial)" ) \
143 for( int counter##3 = 0; counter##3 < counter##Max(3); counter##3++ ) { \
144 tarch::la::Vector<4, int> counter = { counter##0, counter##1, counter##2, counter##3 };
145
146#define simtDforWithSchedulerInstructions5d(counter, max, loopParallelism) \
147 { \
148 tarch::la::Vector<5,int> counter##Max(max); \
149 auto __loopParallelismUsedWithinMacroWithUniqueName = loopParallelism; \
150 _Pragma( "omp parallel for if (__loopParallelismUsedWithinMacroWithUniqueName==::peano4::utils::LoopPlacement::SpreadOut)" ) \
151 for( int counter##0 = 0; counter##0 < counter##Max(0); counter##0++ ) \
152 for( int counter##1 = 0; counter##1 < counter##Max(1); counter##1++ ) \
153 for( int counter##2 = 0; counter##2 < counter##Max(2); counter##2++ ) \
154 for( int counter##3 = 0; counter##3 < counter##Max(3); counter##3++ ) { \
155 _Pragma( "omp simd if (__loopParallelismUsedWithinMacroWithUniqueName!=::peano4::utils::LoopPlacement::Serial)" ) \
156 for( int counter##4 = 0; counter##4 < counter##Max(4); counter##4++ ) { \
157 tarch::la::Vector<5, int> counter = { counter##0, counter##1, counter##2, counter##3, counter##4 };
158
159#define endSimtDfor }}}