提交 f9788298 authored 作者: Frederic's avatar Frederic 提交者: Arnaud Bergeron

better error message

上级 b2d72c7c
...@@ -363,11 +363,19 @@ CudaNdarray_HOST_DIMS(%(var)s)[2], ...@@ -363,11 +363,19 @@ CudaNdarray_HOST_DIMS(%(var)s)[2],
CudaNdarray_HOST_DIMS(%(var)s)[3] CudaNdarray_HOST_DIMS(%(var)s)[3]
); );
if (%(err)s != CUDNN_STATUS_SUCCESS) { if (%(err)s != CUDNN_STATUS_SUCCESS) {
PyErr_Format(PyExc_RuntimeError, "could not set filter descriptor: %%s", PyErr_Format(PyExc_RuntimeError,
cudnnGetErrorString(%(err)s)); "%(cls)s: could not set filter descriptor: %%s."
" dims= %%d %%d %%d %%d",
cudnnGetErrorString(%(err)s),
CudaNdarray_HOST_DIMS(%(var)s)[0],
CudaNdarray_HOST_DIMS(%(var)s)[1],
CudaNdarray_HOST_DIMS(%(var)s)[2],
CudaNdarray_HOST_DIMS(%(var)s)[3]);
%(fail)s %(fail)s
} }
""" % dict(var=var, desc=desc, err=err, fail=fail) """ % dict(var=var, desc=desc, err=err, fail=fail,
cls=self.__class__.__name__)
def c_set_tensor4d(self, *arg): def c_set_tensor4d(self, *arg):
return c_set_tensor4d(*arg) return c_set_tensor4d(*arg)
...@@ -396,7 +404,7 @@ if (!CudaNdarray_is_c_contiguous(%s)) { ...@@ -396,7 +404,7 @@ if (!CudaNdarray_is_c_contiguous(%s)) {
sub['fail']) sub['fail'])
return """ return """
cudnnStatus_t err%(name)s; cudnnStatus_t err%(name)s = CUDNN_STATUS_SUCCESS;
%(checks)s %(checks)s
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论