From 90d51c5f897ed7f6230e9280a865b096a19a11b6 Mon Sep 17 00:00:00 2001
From: Stan Seibert <stan@mtrr.org>
Date: Mon, 2 Jan 2012 12:53:33 -0500
Subject: Improve identification of CUDA vector types in struct builder

---
 chroma/gpu/tools.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/chroma/gpu/tools.py b/chroma/gpu/tools.py
index 98df7ce..707a45d 100644
--- a/chroma/gpu/tools.py
+++ b/chroma/gpu/tools.py
@@ -160,7 +160,7 @@ def make_gpu_struct(size, members):
 
             cuda.memcpy_htod(int(struct)+i, np.intp(int(member)))
             i += 8
-        elif np.isscalar(member) or hasattr(member, 'dtype') and member.dtype in vec_dtypes:
+        elif np.isscalar(member) or (hasattr(member, 'dtype') and member.dtype in vec_dtypes and member.shape == ()):
             cuda.memcpy_htod(int(struct)+i, member)
             i += member.nbytes
         else:
-- 
cgit