提交 410016f9 authored 作者: Frederic Bastien's avatar Frederic Bastien

Fix memory leak in error handling.

上级 c050562f
...@@ -328,6 +328,7 @@ PyObject * CudaNdarray_CreateArrayObj(CudaNdarray * self) ...@@ -328,6 +328,7 @@ PyObject * CudaNdarray_CreateArrayObj(CudaNdarray * self)
if (CUBLAS_STATUS_SUCCESS != cublasGetError()) if (CUBLAS_STATUS_SUCCESS != cublasGetError())
{ {
PyErr_SetString(PyExc_RuntimeError, "error copying data to host"); PyErr_SetString(PyExc_RuntimeError, "error copying data to host");
Py_DECREF(contiguous_self);
Py_DECREF(rval); Py_DECREF(rval);
rval = NULL; rval = NULL;
} }
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论