feat(mge): support python -m megengine.distributed.server

GitOrigin-RevId: f1e5c8e3cf441248157060efc04b8ff9a19b154b

imperative/python/megengine/distributed/server.py

@@ -252,3 +252,22 @@ class Client:
     def user_get(self, key):
         """Get user defined key-value pairs across processes."""
         return self.proxy.user_get(key)
+
+
+def main(port=0, verbose=True):
+    mm_server_port = create_mm_server("0.0.0.0", 0)
+    server = ThreadXMLRPCServer(("0.0.0.0", port), logRequests=verbose)
+    server.register_instance(Methods(mm_server_port))
+    _, port = server.server_address
+    print("serving on port", port)
+    server.serve_forever()
+
+
+if __name__ == "__main__":
+    import argparse
+
+    ap = argparse.ArgumentParser()
+    ap.add_argument("-p", "--port", type=int, default=0)
+    ap.add_argument("-v", "--verbose", type=bool, default=True)
+    args = ap.parse_args()
+    main(port=args.port, verbose=args.verbose)