Update solve.py

loganbvh · Sep 18, 2023 · 0832ca2 · 0832ca2
1 parent eea0d11
commit 0832ca2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tdgl/solver/solve.py b/tdgl/solver/solve.py
@@ -369,6 +369,7 @@ def update(
                 get_A_induced_cupy[128, 1024](
                     J_site, areas, sites, edge_centers, new_A_induced
                 )
+                cupy.cuda.Stream.null.synchronize()
             else:
                 new_A_induced = get_A_induced_numba(J_site, areas, sites, edge_centers)
             # Update induced vector potential using Polyak's method