提交 0749674e authored 作者: abergeron's avatar abergeron

Merge pull request #2004 from nouiz/crash_mrg

[CRASH] Fix compilation crash in mrg on the GPU.
......@@ -712,7 +712,7 @@ class GPU_mrg_uniform(mrg_uniform_base, GpuOp):
}
}
if (PyArray_NDIM(%(o_rstate)s) != 1)
if (CudaNdarray_NDIM(%(o_rstate)s) != 1)
{
PyErr_SetString(PyExc_ValueError, "rstate must be vector");
%(fail)s;
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论