提交 34260d1a authored 作者: Frederic's avatar Frederic

make filter for CudaNdarray check for stride == 0 for dim == 1.

上级 d94978bc
...@@ -2956,6 +2956,12 @@ filter(PyObject* __unsed_self, PyObject *args) // args = (data, broadcastable, s ...@@ -2956,6 +2956,12 @@ filter(PyObject* __unsed_self, PyObject *args) // args = (data, broadcastable, s
Py_DECREF(py_data); Py_DECREF(py_data);
Py_DECREF(broadcastable); Py_DECREF(broadcastable);
return NULL; return NULL;
}else if (CudaNdarray_HOST_DIMS(cnda)[i] == 1 && CudaNdarray_HOST_STRIDES(cnda)[i] != 0){
PyErr_Format(PyExc_TypeError, "Non-zeros strides(%d) on dimension %d of size 1",
CudaNdarray_HOST_STRIDES(cnda)[i], i);
Py_DECREF(py_data);
Py_DECREF(broadcastable);
return NULL;
} }
} }
Py_DECREF(broadcastable); Py_DECREF(broadcastable);
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论