提交 fe69a5a5 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Fix another potential memory leak.

上级 b6a058b7
...@@ -636,7 +636,9 @@ PyObject * CudaNdarray_Reshape(CudaNdarray * self, PyObject * shape) ...@@ -636,7 +636,9 @@ PyObject * CudaNdarray_Reshape(CudaNdarray * self, PyObject * shape)
} }
if (rval_size==0) if (rval_size==0)
{ {
return CudaNdarray_NewDims(rval_nd, rval_dims); CudaNdarray * rval = CudaNdarray_NewDims(rval_nd, rval_dims);
free(rval_dims);
return (PyObject*) rval;
} }
if(CudaNdarray_is_c_contiguous(self)) if(CudaNdarray_is_c_contiguous(self))
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论