Fix for GPU

xsuite · Oct 21, 2022 · 89a0bf2 · 89a0bf2
1 parent 8a994b8
commit 89a0bf2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_beambeam3d.py b/tests/test_beambeam3d.py
@@ -621,6 +621,7 @@ def test_beambeam3d_old_interface():
             part_before_tracking = part.copy()
             part.move(_context=context)
             bb.track(part)
+            part.move(_context=xo.ContextCpu())
 
             for cc in 'x px y py zeta delta'.split():
                 val_test = getattr(part, cc)[0]