1 files changed, 1 insertions, 0 deletions
diff --git a/view.py b/view.py
index c827442..3958e05 100755
--- a/view.py
+++ b/view.py
@@ -82,6 +82,7 @@ def view(geometry, name=''):
     def render():
         """Render the mesh and display to screen."""
         cuda_raytrace(np.int32(pixels.size), origins_gpu, directions_gpu, np.int32(geometry.node_map.size-1), np.int32(geometry.first_node), pixels_gpu, texrefs=texrefs, **gpu_kwargs)
+        cuda.Context.synchronize()
 
         cuda.memcpy_dtoh(pixels, pixels_gpu)
         pygame.surfarray.blit_array(screen, pixels.reshape(size))