提交 14241a7d authored 作者: Amjad Almahairi's avatar Amjad Almahairi

make sure we don't get out of memory

上级 f408152d
...@@ -314,9 +314,9 @@ class MultinomialWOReplacementFromUniform(MultinomialFromUniform): ...@@ -314,9 +314,9 @@ class MultinomialWOReplacementFromUniform(MultinomialFromUniform):
*z_nc = m; *z_nc = m;
// renormalize the nth row of pvals, reuse (cummul-*pvals_nm) to initialize the sum // renormalize the nth row of pvals, reuse (cummul-*pvals_nm) to initialize the sum
dtype_%(pvals)s sum = cummul - *pvals_nm; dtype_%(pvals)s sum = cummul - *pvals_nm;
dtype_%(pvals)s* pvals_n = (dtype_%(pvals)s*)PyArray_GETPTR2(pvals_copy, n, m+1); dtype_%(pvals)s* pvals_n = (dtype_%(pvals)s*)PyArray_GETPTR2(pvals_copy, n, m);
*pvals_nm = 0.; *pvals_nm = 0.;
for (int k = m+1; k < nb_outcomes; ++k) for (int k = m; k < nb_outcomes; ++k)
{ {
sum = sum + *pvals_n; sum = sum + *pvals_n;
pvals_n++; pvals_n++;
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论