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

Fix memory leak in C code for blocksparse.

上级 8f9c2a12
......@@ -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,
W=W, fail=sub['fail'])
def c_code_cache_version(self):
return (0,)
return (1,)
def grad(self, inputs, grads):
o, W, h, inputIdx, outputIdx = inputs
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论