提交 7aec2f4c authored 作者: Frederic's avatar Frederic

Fix compilation crash in mrg on the GPU.

上级 f15ef349
...@@ -712,7 +712,7 @@ class GPU_mrg_uniform(mrg_uniform_base, GpuOp): ...@@ -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"); PyErr_SetString(PyExc_ValueError, "rstate must be vector");
%(fail)s; %(fail)s;
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论