提交 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):
height, width,
c, d, step_x, step_y,
grid_c, grid_d,
CudaNdarray_HOST_STRIDES(%(ten4)s)[0],
CudaNdarray_HOST_STRIDES(%(ten4)s)[1],
CudaNdarray_HOST_STRIDES(%(ten4)s)[2],
CudaNdarray_HOST_STRIDES(%(ten4)s)[3],
PyGpuArray_STRIDES(%(ten4)s)[0],
PyGpuArray_STRIDES(%(ten4)s)[1],
PyGpuArray_STRIDES(%(ten4)s)[2],
PyGpuArray_STRIDES(%(ten4)s)[3],
CudaNdarray_DEV_DATA(%(ten4)s),
CudaNdarray_HOST_STRIDES(%(z)s)[0],
CudaNdarray_HOST_STRIDES(%(z)s)[1],
PyGpuArray_STRIDES(%(z)s)[0],
PyGpuArray_STRIDES(%(z)s)[1],
CudaNdarray_DEV_DATA(%(z)s)
);
CNDA_THREAD_SYNC;
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论