From e7f85d274abc47d47ec63670189809593f229052 Mon Sep 17 00:00:00 2001 From: "Colin L. Rice (Meta Employee)" Date: Thu, 9 Jan 2025 19:43:10 -0800 Subject: [PATCH] easy: sort dictionary keys for inductor config when publishing (#143307) Summary: This means we should get consistent logging strings for the same config on different ranks X-link: https://github.com/pytorch/pytorch/pull/143307 Approved by: https://github.com/xmfan Reviewed By: clee2000 Differential Revision: D67985940 Pulled By: c00w fbshipit-source-id: 0754fcc6d49b5768ffc34f12b1e693c65624cda8 --- userbenchmark/dynamo/dynamobench/_dynamo/utils.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/userbenchmark/dynamo/dynamobench/_dynamo/utils.py b/userbenchmark/dynamo/dynamobench/_dynamo/utils.py index 5cbba76a9a..d5c2206392 100644 --- a/userbenchmark/dynamo/dynamobench/_dynamo/utils.py +++ b/userbenchmark/dynamo/dynamobench/_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