rng_mrg - use numpy.seterr to suppress overflow warnings

theano/sandbox/rng_mrg.py

@@ -31,11 +31,15 @@ def mulmod(a, b, c, m):
 
 def matVecModM(A, s, m):
     # return (A * s) % m
-    x = numpy.zeros_like(s)
-    for i in xrange(len(x)):
-        for j in xrange(len(s)):
-            x[i] = mulmod(A[i][j], s[j], x[i], m)
-    return x
+    err_orig = numpy.seterr(over='ignore')
+    try:
+        x = numpy.zeros_like(s)
+        for i in xrange(len(x)):
+            for j in xrange(len(s)):
+                x[i] = mulmod(A[i][j], s[j], x[i], m)
+        return x
+    finally:
+        numpy.seterr(**err_orig)
 
 def multMatVect(v, A, m1, B, m2):
     #multiply the first half of v by A with a modulo of m1
@@ -81,54 +85,58 @@ def ff_2p72(rstate):
     return multMatVect(rstate, A1p72, M1, A2p72, M2)
 
 def mrg_next_value(rstate, new_rstate):
-    x11, x12, x13, x21, x22, x23 = rstate
-    assert type(x11) == numpy.int32
-
-    i0, i7, i9, i15, i16, i22, i24 = [numpy.int32(i)
-            for i in (0,7, 9, 15, 16, 22, 24)]
-
-    #first component
-    y1 = (((x12 & MASK12) << i22) + (x12 >> i9)
-        + ((x13 & MASK13) << i7) + (x13 >> i24))
-
-    assert type(y1) == numpy.int32
-    if (y1 < 0 or y1 >= M1):     #must also check overflow
-        y1 -= M1;
-    y1 += x13;
-    if (y1 < 0 or y1 >= M1):
-        y1 -= M1;
-
-    x13 = x12;
-    x12 = x11;
-    x11 = y1;
-
-    #second component
-    y1 = ((x21 & MASK2) << i15) + (MULT2 * (x21 >> i16));
-    assert type(y1) == numpy.int32
-    if (y1 < 0 or y1 >= M2):
-        y1 -= M2;
-    y2 = ((x23 & MASK2) << i15) + (MULT2 * (x23 >> i16));
-    assert type(y2) == numpy.int32
-    if (y2 < 0 or y2 >= M2):
-        y2 -= M2;
-    y2 += x23;
-    if (y2 < 0 or y2 >= M2):
-        y2 -= M2;
-    y2 += y1;
-    if (y2 < 0 or y2 >= M2):
-        y2 -= M2;
-
-    x23 = x22;
-    x22 = x21;
-    x21 = y2;
-
-    # Must never return either 0 or M1+1
-    new_rstate[...] = [x11, x12, x13, x21, x22, x23]
-    assert new_rstate.dtype == numpy.int32
-    if (x11 <= x21):
-        return (x11 - x21 + M1) * NORM
-    else:
-        return (x11 - x21) * NORM
+    err_orig = numpy.seterr(over='ignore')
+    try:
+        x11, x12, x13, x21, x22, x23 = rstate
+        assert type(x11) == numpy.int32
+
+        i0, i7, i9, i15, i16, i22, i24 = [numpy.int32(i)
+                for i in (0,7, 9, 15, 16, 22, 24)]
+
+        #first component
+        y1 = (((x12 & MASK12) << i22) + (x12 >> i9)
+            + ((x13 & MASK13) << i7) + (x13 >> i24))
+
+        assert type(y1) == numpy.int32
+        if (y1 < 0 or y1 >= M1):     #must also check overflow
+            y1 -= M1;
+        y1 += x13;
+        if (y1 < 0 or y1 >= M1):
+            y1 -= M1;
+
+        x13 = x12;
+        x12 = x11;
+        x11 = y1;
+
+        #second component
+        y1 = ((x21 & MASK2) << i15) + (MULT2 * (x21 >> i16));
+        assert type(y1) == numpy.int32
+        if (y1 < 0 or y1 >= M2):
+            y1 -= M2;
+        y2 = ((x23 & MASK2) << i15) + (MULT2 * (x23 >> i16));
+        assert type(y2) == numpy.int32
+        if (y2 < 0 or y2 >= M2):
+            y2 -= M2;
+        y2 += x23;
+        if (y2 < 0 or y2 >= M2):
+            y2 -= M2;
+        y2 += y1;
+        if (y2 < 0 or y2 >= M2):
+            y2 -= M2;
+
+        x23 = x22;
+        x22 = x21;
+        x21 = y2;
+
+        # Must never return either 0 or M1+1
+        new_rstate[...] = [x11, x12, x13, x21, x22, x23]
+        assert new_rstate.dtype == numpy.int32
+        if (x11 <= x21):
+            return (x11 - x21 + M1) * NORM
+        else:
+            return (x11 - x21) * NORM
+    finally:
+        numpy.seterr(**err_orig)
 
 class mrg_uniform_base(Op):
     def __init__(self, output_type, inplace=False):