提交 69f68bea authored 作者: Vincent Dumoulin's avatar Vincent Dumoulin

Change x->nd to PyGpuArray_NDIM(x)

上级 e6fd3d5b
...@@ -100,17 +100,17 @@ class GpuCrossentropySoftmaxArgmax1HotWithBias(Op): ...@@ -100,17 +100,17 @@ class GpuCrossentropySoftmaxArgmax1HotWithBias(Op):
fail = sub['fail'] fail = sub['fail']
sio = StringIO() sio = StringIO()
print >> sio, """ print >> sio, """
if (%(y_idx)s->nd != 1) if (PyGpuArray_NDIM(%(y_idx)s) != 1)
{ {
PyErr_SetString(PyExc_ValueError, "y_idx not 1d tensor"); PyErr_SetString(PyExc_ValueError, "y_idx not 1d tensor");
%(fail)s; %(fail)s;
} }
if (%(x)s->nd != 2) if (PyGpuArray_NDIM(%(x)s) != 2)
{ {
PyErr_SetString(PyExc_ValueError, "x not 2d tensor"); PyErr_SetString(PyExc_ValueError, "x not 2d tensor");
%(fail)s; %(fail)s;
} }
if (%(b)s->nd != 1) if (PyGpuArray_NDIM(%(b)s) != 1)
{ {
PyErr_SetString(PyExc_ValueError, "b not 1d tensor"); PyErr_SetString(PyExc_ValueError, "b not 1d tensor");
%(fail)s; %(fail)s;
...@@ -262,9 +262,9 @@ class GpuCrossentropySoftmax1HotWithBiasDx(Op): ...@@ -262,9 +262,9 @@ class GpuCrossentropySoftmax1HotWithBiasDx(Op):
dx, = out dx, = out
fail = sub['fail'] fail = sub['fail']
return """ return """
if ((%(dnll)s->nd != 1) if ((PyGpuArray_NDIM(%(dnll)s) != 1)
|| (%(sm)s->nd != 2) || (PyGpuArray_NDIM(%(sm)s) != 2)
|| (%(y_idx)s->nd != 1)) || (PyGpuArray_NDIM(%(y_idx)s) != 1))
{ {
PyErr_SetString(PyExc_ValueError, "rank error"); PyErr_SetString(PyExc_ValueError, "rank error");
%(fail)s; %(fail)s;
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论