提交 df512c7f authored 作者: Frederic's avatar Frederic

Use numpy new interface.

上级 5be14f49
...@@ -206,12 +206,12 @@ class GpuImages2Neibs(Images2Neibs, GpuOp): ...@@ -206,12 +206,12 @@ class GpuImages2Neibs(Images2Neibs, GpuOp):
int grid_d = -1; int grid_d = -1;
{ {
if (%(ten4)s->nd != 4) if (CudaNdarray_NDIM(%(ten4)s) != 4)
{ {
PyErr_Format(PyExc_TypeError, "pvals wrong rank"); PyErr_Format(PyExc_TypeError, "pvals wrong rank");
%(fail)s; %(fail)s;
} }
if (%(neib_shape)s->nd != 1) if (PyArray_NDIM(%(neib_shape)s) != 1)
{ {
PyErr_Format(PyExc_TypeError, "unis wrong rank"); PyErr_Format(PyExc_TypeError, "unis wrong rank");
%(fail)s; %(fail)s;
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论