compiler: Patch GuardBoundNext pickling

devito/ir/support/guards.py

@@ -103,7 +103,7 @@ class GuardBoundGt(BaseGuardBound, Gt):
 # *** GuardBoundNext
 
 
-class BaseGuardBoundNext(Guard):
+class BaseGuardBoundNext(Guard, Pickable):
 
     """
     A guard to avoid out-of-bounds iteration.
@@ -118,6 +118,8 @@ class BaseGuardBoundNext(Guard):
     given `direction`.
     """
 
+    __rargs__ = ('d', 'direction')
+
     def __new__(cls, d, direction, **kwargs):
         assert isinstance(d, Dimension)
         assert isinstance(direction, IterationDirection)

tests/test_pickle.py

@@ -10,7 +10,7 @@
                     Dimension, SubDimension, ConditionalDimension, IncrDimension,
                     TimeDimension, SteppingDimension, Operator, MPI, Min, solve,
                     PrecomputedSparseTimeFunction)
-from devito.ir import GuardFactor
+from devito.ir import Backward, GuardFactor, GuardBound, GuardBoundNext
 from devito.data import LEFT, OWNED
 from devito.mpi.halo_scheme import Halo
 from devito.mpi.routines import (MPIStatusObject, MPIMsgEnriched, MPIRequestObject,
@@ -388,6 +388,28 @@ def test_guard_factor(self, pickle):
 
         assert str(gf) == str(new_gf)
 
+    def test_guard_bound(self, pickle):
+        d = Dimension(name='d')
+
+        gb = GuardBound(d, 3)
+
+        pkl_gb = pickle.dumps(gb)
+        new_gb = pickle.loads(pkl_gb)
+
+        assert str(gb) == str(new_gb)
+
+    def test_guard_bound_next(self, pickle):
+        d = Dimension(name='d')
+        cd = ConditionalDimension(name='cd', parent=d, factor=4)
+
+        for i in [d, cd]:
+            gbn = GuardBoundNext(i, Backward)
+
+            pkl_gbn = pickle.dumps(gbn)
+            new_gbn = pickle.loads(pkl_gbn)
+
+            assert str(gbn) == str(new_gbn)
+
     def test_temp_function(self, pickle):
         grid = Grid(shape=(3, 3))
         d = Dimension(name='d')