[c10d][Logging] Remove args and kwargs from c10d logging (pytorch#140169)

This PR is trying to reland pytorch#139804

We now don't want to log args and kwargs directly because if they contain tensor or tensor subclass it would take lots of time in conversion to string or even not supported.

Pull Request resolved: pytorch#140169
Approved by: https://github.com/wz337, https://github.com/kwen2501

Author: fduwjj

test/distributed/test_c10d_logger.py

@@ -118,16 +118,14 @@ def test_exception_logger(self) -> None:
                 re.search("({.+})", captured.output[0]).group(0).replace("'", '"')
             )
 
-            self.assertEqual(len(error_msg_dict), 10)
+            self.assertEqual(len(error_msg_dict), 9)
 
             self.assertIn("pg_name", error_msg_dict.keys())
             self.assertEqual("None", error_msg_dict["pg_name"])
 
             self.assertIn("func_name", error_msg_dict.keys())
             self.assertEqual("broadcast", error_msg_dict["func_name"])
 
-            self.assertIn("args", error_msg_dict.keys())
-
             self.assertIn("backend", error_msg_dict.keys())
             self.assertEqual("nccl", error_msg_dict["backend"])
 
@@ -162,16 +160,14 @@ def test_time_logger(self) -> None:
             msg_dict = json.loads(
                 re.search("({.+})", captured.output[0]).group(0).replace("'", '"')
             )
-            self.assertEqual(len(msg_dict), 10)
+            self.assertEqual(len(msg_dict), 9)
 
             self.assertIn("pg_name", msg_dict.keys())
             self.assertEqual("None", msg_dict["pg_name"])
 
             self.assertIn("func_name", msg_dict.keys())
             self.assertEqual("_dummy_sleep", msg_dict["func_name"])
 
-            self.assertIn("args", msg_dict.keys())
-
             self.assertIn("backend", msg_dict.keys())
             self.assertEqual("nccl", msg_dict["backend"])
 

torch/distributed/c10d_logger.py

@@ -53,7 +53,6 @@ def _get_msg_dict(func_name, *args, **kwargs) -> Dict[str, Any]:
         group = kwargs.get("group") or kwargs.get("process_group")
         msg_dict = {
             "func_name": f"{func_name}",
-            "args": f"{args}, {kwargs}",
             "pg_name": f"{dist._get_process_group_name(kwargs.get('pg'))}",  # type: ignore[arg-type]
             "backend": f"{dist.get_backend(group)}",
             "world_size": f"{dist.get_world_size()}",
@@ -67,7 +66,6 @@ def _get_msg_dict(func_name, *args, **kwargs) -> Dict[str, Any]:
     else:
         msg_dict = {
             "func_name": f"{func_name}",
-            "args": f"{args}, {kwargs}",
         }
     return msg_dict