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