From b63412ff9c45a9b66cda0003fe66aa7e1b626361 Mon Sep 17 00:00:00 2001
From: Hicham Agueny <95568317+HichamAgueny@users.noreply.github.com>
Date: Tue, 20 Feb 2024 15:07:05 +0100
Subject: [PATCH] add hipDeviceSynchronize

---
 GPUSimulators/FORCE.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/GPUSimulators/FORCE.py b/GPUSimulators/FORCE.py
index ebf4416..205c4d7 100644
--- a/GPUSimulators/FORCE.py
+++ b/GPUSimulators/FORCE.py
@@ -187,9 +187,11 @@ class FORCE (Simulator.BaseSimulator):
                         )
                     )
                 )
+            
+        hip_check(hip.hipDeviceSynchronize())    
         self.u0, self.u1 = self.u1, self.u0
             
-        hip_check(hip.hipDeviceSynchronize())
+        
         hip_check(hip.hipModuleUnload(module))
             
         hip_check(hip.hipFree(cfl_data))