提交 bb055f34 authored 作者: Olivier Delalleau's avatar Olivier Delalleau

Fixed crash due to integer division, and got rid of gcc warning in C code

上级 e4c45d5a
...@@ -263,7 +263,7 @@ class mrg_uniform(mrg_uniform_base): ...@@ -263,7 +263,7 @@ class mrg_uniform(mrg_uniform_base):
if (%(size)s->dimensions[0] != %(ndim)s) if (%(size)s->dimensions[0] != %(ndim)s)
{ {
PyErr_Format(PyExc_ValueError, "size must have length %%i (not %%i)", PyErr_Format(PyExc_ValueError, "size must have length %%i (not %%i)",
%(ndim)s, %(size)s->dimensions[0]); %(ndim)s, int(%(size)s->dimensions[0]));
%(fail)s %(fail)s
} }
if (%(size)s->descr->type_num != PyArray_INT32) if (%(size)s->descr->type_num != PyArray_INT32)
...@@ -789,11 +789,11 @@ class MRG_RandomStreams(object): ...@@ -789,11 +789,11 @@ class MRG_RandomStreams(object):
flattened = self.uniform(size=(n_samples,), dtype=dtype) flattened = self.uniform(size=(n_samples,), dtype=dtype)
if constant: if constant:
U1 = flattened[:n_samples/2] U1 = flattened[:n_samples // 2]
U2 = flattened[n_samples/2:] U2 = flattened[n_samples // 2:]
else: else:
U1 = flattened[:prod(flattened.shape)/2] U1 = flattened[:prod(flattened.shape) // 2]
U2 = flattened[prod(flattened.shape)/2:] U2 = flattened[prod(flattened.shape) // 2:]
#normal_samples = zeros_like(flattened) #normal_samples = zeros_like(flattened)
sqrt_ln_U1 = sqrt(-2.0*log(U1)) sqrt_ln_U1 = sqrt(-2.0*log(U1))
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论