From c4a03f39247967c9c55285be90cb72abeed0c8ed Mon Sep 17 00:00:00 2001
From: Henry <henry.korb@geo.uu.se>
Date: Thu, 12 Oct 2023 15:48:33 +0200
Subject: [PATCH] properly reset body force

---
 .../LBM/GPUHelperFunctions/RunCollision.cuh               | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/gpu/VirtualFluids_GPU/LBM/GPUHelperFunctions/RunCollision.cuh b/src/gpu/VirtualFluids_GPU/LBM/GPUHelperFunctions/RunCollision.cuh
index ba1034ac5..9c5b17adf 100644
--- a/src/gpu/VirtualFluids_GPU/LBM/GPUHelperFunctions/RunCollision.cuh
+++ b/src/gpu/VirtualFluids_GPU/LBM/GPUHelperFunctions/RunCollision.cuh
@@ -58,9 +58,9 @@ __global__ void runCollision(CollisionFunctor collision, GPUCollisionParameter c
         para.forceZ = (collisionParameter.forces[2] + collisionParameter.bodyForceZ[k_000]) * c1o2 * collisionParameter.forceFactor;
 
         // Reset body force. To be used when not using round-off correction.
-        collisionParameter.bodyForceX[k_000] = 0.0f;
-        collisionParameter.bodyForceX[k_000] = 0.0f;
-        collisionParameter.bodyForceX[k_000] = 0.0f;
+        collisionParameter.bodyForceX[k_000] = c0o1;
+        collisionParameter.bodyForceY[k_000] = c0o1;
+        collisionParameter.bodyForceZ[k_000] = c0o1;
 
         ////////////////////////////////////////////////////////////////////////////////////
         //!> Round-off correction
@@ -97,7 +97,7 @@ __global__ void runCollision(CollisionFunctor collision, GPUCollisionParameter c
     vf::lbm::MacroscopicValues macroscopicValues;
     collision(para, macroscopicValues, turbulentViscosity);
 
-    if (writeMacroscopicVariables) {
+    if (writeMacroscopicVariables || turbulenceModel == vf::lbm::TurbulenceModel::AMD) {
         collisionParameter.vx[k_000] = macroscopicValues.vx;
         collisionParameter.vy[k_000] = macroscopicValues.vy;
         collisionParameter.vz[k_000] = macroscopicValues.vz;
-- 
GitLab