提交 c35ef4d8 authored 作者: abergeron's avatar abergeron 提交者: GitHub

Merge pull request #5996 from xiaoqie/gpureduce-fix

Fix memory leak in GpuCAReduceCPY
...@@ -2766,6 +2766,7 @@ class GpuCAReduceCPY(GpuKernelBase, HideC, CAReduceDtype): ...@@ -2766,6 +2766,7 @@ class GpuCAReduceCPY(GpuKernelBase, HideC, CAReduceDtype):
if (%(cast_out)d) { if (%(cast_out)d) {
err = GpuArray_move(&%(output)s->ga, &tmp->ga); err = GpuArray_move(&%(output)s->ga, &tmp->ga);
Py_XDECREF(tmp);
if (err != GA_NO_ERROR) { if (err != GA_NO_ERROR) {
PyErr_Format(PyExc_RuntimeError, PyErr_Format(PyExc_RuntimeError,
"gpuarray error: GpuCAReduceCPY [cast]: %%s.", "gpuarray error: GpuCAReduceCPY [cast]: %%s.",
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论