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

Use numpy new interface.

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