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

Convert x->nd to PyGpuArray_NDIM(x) in op GpuImages2Neibs

上级 85023583
...@@ -210,12 +210,12 @@ class GpuImages2Neibs(Images2Neibs, Op): ...@@ -210,12 +210,12 @@ class GpuImages2Neibs(Images2Neibs, Op):
int grid_d = -1; int grid_d = -1;
{ {
if (%(ten4)s->nd != 4) if (PyGpuArray_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 (PyGpuArray_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 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论