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

Fix reference count for matrix copy

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