提交 656055f6 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Plug memory leak in GpuDot22.

上级 6c5ab287
......@@ -239,6 +239,7 @@ class GpuDot22(BlasOp, Dot22):
dims[0] = PyGpuArray_DIMS(%(A)s)[0];
dims[1] = PyGpuArray_DIMS(%(B)s)[1];
Py_XDECREF(%(out)s);
%(out)s = pygpu_empty(2, dims,
%(typecode)s,
GA_C_ORDER,
......@@ -262,7 +263,7 @@ class GpuDot22(BlasOp, Dot22):
return code
def c_code_cache_version(self):
return (1,)
return (2,)
def c_headers(self):
ret = super(GpuDot22, self).c_headers()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论