rambo wip

Author: AllanZyne

examples/rambo.py

@@ -2,6 +2,7 @@
 import numpy as np
 
 def sp_rambo(sp_C1, sp_F1, sp_Q1, sp_output, C1, F1, Q1, output):
+
     sp_C = 2.0 * sp_C1 - 1.0
     sp_S = sp.sqrt(1 - sp.square(sp_C))
     sp_F = 2.0 * sp.pi * sp_F1
@@ -22,11 +23,7 @@ def sp_rambo(sp_C1, sp_F1, sp_Q1, sp_output, C1, F1, Q1, output):
     output[:, :, 2] = Q * S * np.cos(F)
     output[:, :, 3] = Q * C
 
-    # assert np.allclose(sp.to_numpy(sp_C), C)
-    # assert np.allclose(sp.to_numpy(sp_S), S)
-    # assert np.allclose(sp.to_numpy(sp_F), F)
-    # assert np.allclose(sp.to_numpy(sp_Q), Q)
-    # print(sp_C, sp_S, sp_F, sp_Q)
+    sp.sync()
 
 
 def sp_initialize(nevts, nout, seed, types_dict):
@@ -37,6 +34,8 @@ def sp_initialize(nevts, nout, seed, types_dict):
     F1 = sp.random.rand(nevts, nout)
     Q1 = sp.random.rand(nevts, nout) * sp.random.rand(nevts, nout)
 
+    sp.sync()
+
     return (C1, F1, Q1, sp.zeros((nevts, nout, 4), dtype))
 
 def np_rambo(nevts, nout, C1, F1, Q1, output):