提交 0299784f authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Ensure RNG_MRG always has contiguous output

上级 8002842c
......@@ -499,6 +499,7 @@ class GPU_mrg_uniform(mrg_uniform_base, GpuOp):
int n_streams, n_streams_used_in_this_call;
int must_alloc_sample = ((NULL == %(o_sample)s)
|| !CudaNdarray_Check(py_%(o_sample)s)
|| !CudaNdarray_is_c_contiguous(%(o_sample)s)
|| (%(o_sample)s->nd != %(ndim)s));
if (%(size)s->nd != 1)
......@@ -593,7 +594,7 @@ class GPU_mrg_uniform(mrg_uniform_base, GpuOp):
//////// </ code generated by mrg_uniform>
""" %locals()
def c_code_cache_version(self):
return (5,)
return (6,)
def guess_n_streams(size, warn=True):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论