提交 ad61c7e4 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Return an error when there is an error in dnn_pool.c.

上级 4e8eac00
...@@ -94,6 +94,7 @@ int APPLY_SPECIFIC(dnn_pool)(PyGpuArrayObject *img, ...@@ -94,6 +94,7 @@ int APPLY_SPECIFIC(dnn_pool)(PyGpuArrayObject *img,
if (err != CUDNN_STATUS_SUCCESS) { if (err != CUDNN_STATUS_SUCCESS) {
PyErr_Format(PyExc_RuntimeError, "could not set op descriptor %s", cudnnGetErrorString(err)); PyErr_Format(PyExc_RuntimeError, "could not set op descriptor %s", cudnnGetErrorString(err));
return 1;
} }
{ {
......
File mode changed from 100755 to 100644
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论