提交 c774e32e authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix memory leak in C code for blocksparse.

上级 8f9c2a12
...@@ -206,12 +206,15 @@ class SparseBlockGemvSS(GpuOp): ...@@ -206,12 +206,15 @@ class SparseBlockGemvSS(GpuOp):
} }
} }
} }
Py_DECREF(W_part);
Py_DECREF(h_part);
Py_DECREF(out_part);
} }
""" % dict(out=out, h=h, o=o, inputIdx=inputIdx, outputIdx=outputIdx, """ % dict(out=out, h=h, o=o, inputIdx=inputIdx, outputIdx=outputIdx,
W=W, fail=sub['fail']) W=W, fail=sub['fail'])
def c_code_cache_version(self): def c_code_cache_version(self):
return (0,) return (1,)
def grad(self, inputs, grads): def grad(self, inputs, grads):
o, W, h, inputIdx, outputIdx = inputs o, W, h, inputIdx, outputIdx = inputs
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论