reshape float3 array returned by to_float3() in gpu.py

1 files changed, 1 insertions, 1 deletions

diff --git a/gpu.py b/gpu.py
index 4d2197b..e336dac 100644
--- a/gpu.py
+++ b/gpu.py
@@ -19,7 +19,7 @@ import event
 cuda.init()
 
 def to_float3(arr):
-    return arr.astype(np.float32).view(gpuarray.vec.float3)
+    return arr.astype(np.float32).view(gpuarray.vec.float3)[:,0]
 
 def boolean_argsort(condition):
     '''Returns two arrays of indicies indicating which elements of the