提交 a869aa21 authored 作者: f0k's avatar f0k

GpuCorrMM: documented output refcounting

上级 709a3520
......@@ -433,6 +433,9 @@ CudaNdarray* corrMM(CudaNdarray *const bottom,
// Free temporary columns
Py_DECREF(col);
// Note that we don't change the refcount of the output matrix here. Output
// (re)allocation and refcounting is done in BaseGpuCorrMM.c_code_helper();
// in here output is just aliased to one of bottom, weights, or top.
return output;
}
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论