Correct typo. Add stoarge_map to PerformLinker.make_thunk(). Now it works!

theano/compile/function_module.py

@@ -602,7 +602,7 @@ class Function(object):
             storage_map = self.fn.storage_map
             for key in storage_map.keys():
                 if not isinstance(key, theano.tensor.Constant) and \
-                    equiv.has_key(key):
+                    memo.has_key(key):
                     new_storage_map[memo[key]] = storage_map[key]
 
             # copy input storages and use new storage_map to link function

theano/gof/link.py

@@ -663,9 +663,10 @@ class LocalLinker(Linker):
 
     """
 
-    def make_thunk(self, input_storage=None, output_storage=None):
+    def make_thunk(self, input_storage=None, output_storage=None, storage_map=None):
         return self.make_all(input_storage=input_storage,
-                             output_storage=output_storage)[:3]
+                             output_storage=output_storage,
+                             storage_map=storage_map)[:3]
 
     def make_all(self, input_storage, output_storage):
         # By convention, subclasses of LocalLinker should implement this function!