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

Convert CudaNdarray_HOST_DIMSSTRIDES(x) to PyGpuArray_STRIDES(x) in op GpuImages2Neibs

上级 a91c0116
...@@ -378,13 +378,13 @@ class GpuImages2Neibs(Images2Neibs, Op): ...@@ -378,13 +378,13 @@ class GpuImages2Neibs(Images2Neibs, Op):
height, width, height, width,
c, d, step_x, step_y, c, d, step_x, step_y,
grid_c, grid_d, grid_c, grid_d,
CudaNdarray_HOST_STRIDES(%(ten4)s)[0], PyGpuArray_STRIDES(%(ten4)s)[0],
CudaNdarray_HOST_STRIDES(%(ten4)s)[1], PyGpuArray_STRIDES(%(ten4)s)[1],
CudaNdarray_HOST_STRIDES(%(ten4)s)[2], PyGpuArray_STRIDES(%(ten4)s)[2],
CudaNdarray_HOST_STRIDES(%(ten4)s)[3], PyGpuArray_STRIDES(%(ten4)s)[3],
CudaNdarray_DEV_DATA(%(ten4)s), CudaNdarray_DEV_DATA(%(ten4)s),
CudaNdarray_HOST_STRIDES(%(z)s)[0], PyGpuArray_STRIDES(%(z)s)[0],
CudaNdarray_HOST_STRIDES(%(z)s)[1], PyGpuArray_STRIDES(%(z)s)[1],
CudaNdarray_DEV_DATA(%(z)s) CudaNdarray_DEV_DATA(%(z)s)
); );
CNDA_THREAD_SYNC; CNDA_THREAD_SYNC;
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论