提交 411b616e authored 作者: Alexander Matyasko's avatar Alexander Matyasko

Fix reference count for matrix copy

上级 eaad913c
...@@ -132,6 +132,7 @@ fail: ...@@ -132,6 +132,7 @@ fail:
magma_free_pinned(work_data); magma_free_pinned(work_data);
if (iwork_data != NULL) if (iwork_data != NULL)
magma_free_cpu(iwork_data); magma_free_cpu(iwork_data);
Py_XDECREF(A);
cuda_exit(c->ctx); cuda_exit(c->ctx);
return res; return res;
} }
...@@ -147,6 +147,7 @@ fail: ...@@ -147,6 +147,7 @@ fail:
magma_free_pinned(tau_data); magma_free_pinned(tau_data);
if (work_data != NULL) if (work_data != NULL)
gpudata_release(work_data); gpudata_release(work_data);
Py_XDECREF(A);
cuda_exit(c->ctx); cuda_exit(c->ctx);
return res; return res;
} }
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论