提交 52be4733 authored 作者: Frederic's avatar Frederic

Fix memory leak in HostFromGpu when allow_gc=False!

fix gh-1154
上级 4021fc9e
...@@ -89,6 +89,7 @@ class HostFromGpu(GpuOp): ...@@ -89,6 +89,7 @@ class HostFromGpu(GpuOp):
out = outputs[0] out = outputs[0]
fail = sub['fail'] fail = sub['fail']
return """ return """
Py_XDECREF(%(out)s);
%(out)s = (PyArrayObject *) CudaNdarray_CreateArrayObj(%(inp)s); %(out)s = (PyArrayObject *) CudaNdarray_CreateArrayObj(%(inp)s);
if(!%(out)s){ if(!%(out)s){
%(fail)s; %(fail)s;
...@@ -96,7 +97,7 @@ class HostFromGpu(GpuOp): ...@@ -96,7 +97,7 @@ class HostFromGpu(GpuOp):
""" % locals() """ % locals()
def c_code_cache_version(self): def c_code_cache_version(self):
return (1,) return (2,)
host_from_gpu = HostFromGpu() host_from_gpu = HostFromGpu()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论