add view_map to MPIRecvWait

theano/tensor/io.py

@@ -174,6 +174,8 @@ class MPIRecvWait(Op):
     def infer_shape(self, node, shapes):
         return [shapes[1]]
 
+    view_map = {0: [1]}
+
 class MPISend(Op):
     """
     An operation to asynchronously Send an array to a remote host using MPI