diff options
-rw-r--r-- | gpu.py | 2 |
1 files changed, 1 insertions, 1 deletions
@@ -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 |