提交 ab74f6b0 authored 作者: Pierre Luc Carrier's avatar Pierre Luc Carrier

Convert CudaNdarray_DEV_DATA(x) to PyGpuArray_DATA(x) in op GpuImages2Neibs

上级 6902fc7c
...@@ -382,10 +382,10 @@ class GpuImages2Neibs(Images2Neibs, Op): ...@@ -382,10 +382,10 @@ class GpuImages2Neibs(Images2Neibs, Op):
PyGpuArray_STRIDES(%(ten4)s)[1], PyGpuArray_STRIDES(%(ten4)s)[1],
PyGpuArray_STRIDES(%(ten4)s)[2], PyGpuArray_STRIDES(%(ten4)s)[2],
PyGpuArray_STRIDES(%(ten4)s)[3], PyGpuArray_STRIDES(%(ten4)s)[3],
CudaNdarray_DEV_DATA(%(ten4)s), PyGpuArray_DATA(%(ten4)s),
PyGpuArray_STRIDES(%(z)s)[0], PyGpuArray_STRIDES(%(z)s)[0],
PyGpuArray_STRIDES(%(z)s)[1], PyGpuArray_STRIDES(%(z)s)[1],
CudaNdarray_DEV_DATA(%(z)s) PyGpuArray_DATA(%(z)s)
); );
CNDA_THREAD_SYNC; CNDA_THREAD_SYNC;
cudaError_t sts = cudaGetLastError(); cudaError_t sts = cudaGetLastError();
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论