Fix memory leak in scan related to the input and output storage of the inner function.

theano/scan_module/scan_op.py

@@ -1104,6 +1104,13 @@ class Scan(PureOp):
                     # little trick that I used
                     outs[idx][0] = outs[idx][0][:-(n_steps - i)]
 
+        # Make sure to release storage if allow_gc is True
+        if getattr(fn, 'allow_gc', False):
+            for i in input_storage:
+                i.storage[0] = None
+            for o in output_storage:
+                o.storage[0] = None
+
         t_call = time.time() - t0_call
         # NOTE: make this match what's in function_module.Function
         # and this little string helps us to find this spot:

theano/scan_module/scan_perform.pyx

@@ -62,7 +62,7 @@ import copy
 
 
 def get_version():
-    return 0.281
+    return 0.282
 
 @cython.boundscheck(False)
 def perform(
@@ -456,6 +456,13 @@ def perform(
                 sh0 = outs[idx][0].shape[0]
                 outs[idx][0] = outs[idx][0][:sh0-(n_steps - i)]
 
+    # Make sure to release storage if allow_gc is True
+    if getattr(fn, 'allow_gc', False):
+        for i in input_storage:
+            i.storage[0] = None
+        for o in output_storage:
+            o.storage[0] = None
+
     t_call = time.time() - t0_call
 
     if hasattr(fnct.maker, 'profile'):

theano/scan_module/scan_perform_ext.py

@@ -16,7 +16,7 @@ _logger = logging.getLogger('theano.scan_module.scan_perform')
 _logger.setLevel(logging.WARN)
 
 
-version = 0.281  # must match constant returned in function get_version()
+version = 0.282  # must match constant returned in function get_version()
 
 need_reload = False