summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--gpu.py2
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