diff options
Diffstat (limited to 'chroma/gpu/tools.py')
| -rw-r--r-- | chroma/gpu/tools.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/chroma/gpu/tools.py b/chroma/gpu/tools.py index 8bf5d81..ac58f45 100644 --- a/chroma/gpu/tools.py +++ b/chroma/gpu/tools.py @@ -157,10 +157,10 @@ def make_gpu_struct(size, members): 'arrange struct member variables in order of ' 'decreasing size.') - cuda.memcpy_htod(int(struct)+i, np.intp(int(member))) + cuda.memcpy_htod(int(struct)+i, np.asarray(np.intp(int(member)))) i += 8 elif np.isscalar(member) or (hasattr(member, 'dtype') and member.dtype in vec_dtypes and member.shape == ()): - cuda.memcpy_htod(int(struct)+i, member) + cuda.memcpy_htod(int(struct)+i, np.asarray(member)) i += member.nbytes else: raise TypeError('expected a GPU device pointer or scalar type.') |