add finialize

python/paddle/fluid/contrib/utils/__init__.py

@@ -18,5 +18,5 @@ from __future__ import print_function
 from . import hdfs_utils
 from .hdfs_utils import *
 
-__all__ = lookup_table_utils.__all__
+#__all__ = lookup_table_utils.__all__
 __all__ = hdfs_utils.__all__

python/paddle/fluid/distributed/helper.py

@@ -44,5 +44,8 @@ class MPIHelper(object):
     def get_hostname(self):
         import socket
         return socket.gethostname()
+
+    def finalize(self):
+        MPI.Finalize()
     
         

python/paddle/fluid/distributed/ps_instance.py

@@ -97,6 +97,7 @@ class PaddlePSInstance(object):
         pass
 
     def finalize(self):
+        self.dh.finalize()
         pass