提交 608e9aef authored 作者: xiaoqie's avatar xiaoqie

Fix memory leak in GpuCAReduceCPY

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