hpcchris/simpar
1
0
Fork 0
forked from eacdamal/simpar
Code Pull requests Activity

Change order of loops

Browse source 
Try to avoid branching in loops as it prevents vectorization

Signed-off-by: Christoph Niethammer <niethammer@hlrs.de>
This commit is contained in:
Christoph Niethammer 2024-03-13 13:46:44 +01:00
parent 39bee2e2e6
commit d6a0d64176
1 changed files with 6 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
sequential.cpp
View file

@ -72,14 +72,13 @@ int main()
*/
for (int i = 2; i < dt; i++)
{
// Loop over spatial steps
for (int k = 1; k < kmax; k++)
{
// Loop over spatial steps for flow
for (int k = 1; k < kmax; k++) {
Q[k][i + 1] = Q[k][i] + h * (alpha * (P[k - 1][i] - P[k][i]) - beta * Q[k][i] * abs(Q[k][i]));
// DO NOT calculate pressure for the last element such as it is given by default
if (k != kmax - 1)
P[k][i + 1] = P[k][i - 2] + h * (gamma * (Q[k][i] - Q[k + 1][i]));
}
// Loop over steps for pressure, DO NOT calculate pressure for the last element as it is given as a boundary condition
for (int k = 1; k < kmax - 1; k++) {
P[k][i + 1] = P[k][i - 2] + h * (gamma * (Q[k][i] - Q[k + 1][i]));
}
}