easy: sort dictionary keys for inductor config when publishing (pytorch#143307)

This means we should get consistent logging strings for the same
config on different ranks

Pull Request resolved: pytorch#143307
Approved by: https://github.com/xmfan

Author: c00w

test/dynamo/test_utils.py

@@ -412,12 +412,12 @@ def test_inductor_config_parsing_non_conforming_items(self, mocked_inductor_conf
             },
         }
         expected = (
-            """{"some": {"name": "Value is not JSON serializable", "some": true}, """
-            """"data": {"name": "Value is not JSON serializable", "some": true}, """
+            """{"data": {"name": "Value is not JSON serializable", "some": true}, """
             """"list": [{"name": "Value is not JSON serializable", "some": true}, """
             """{"name": "Value is not JSON serializable", "some": true}], """
-            """"object": {"name": "Value is not JSON serializable", "some": true, """
-            """"data": {"name": "Value is not JSON serializable", "some": true}}}"""
+            """"object": {"data": {"name": "Value is not JSON serializable", "some": true}, """
+            """"name": "Value is not JSON serializable", "some": true}, """
+            """"some": {"name": "Value is not JSON serializable", "some": true}}"""
         )
         mocked_inductor_config.get_config_copy.return_value = test_mock_config
         inductor_config_json = utils._scrubbed_inductor_config_for_logging()

torch/_dynamo/utils.py

@@ -1241,7 +1241,10 @@ def default(self, o):
                 del inductor_config_copy[key]
             # Stringify Inductor config
             inductor_conf_str = json.dumps(
-                inductor_config_copy, cls=TypeSafeSerializer, skipkeys=True
+                inductor_config_copy,
+                cls=TypeSafeSerializer,
+                skipkeys=True,
+                sort_keys=True,
             )
         except Exception:
             # Don't crash because of runtime logging errors