From 7c14dd456c8f3185eba574748fcd15d67063a55f Mon Sep 17 00:00:00 2001 From: Gal Rotem Date: Tue, 16 Apr 2024 07:43:31 -0700 Subject: [PATCH] add path to replicated error message (#174) Summary: Pull Request resolved: https://github.com/pytorch/torchsnapshot/pull/174 To make it easier to debug replicated issues Reviewed By: JKSenthil Differential Revision: D56174603 fbshipit-source-id: d1183743ff0e770677aed61b26f55a83c8a9e821 --- torchsnapshot/partitioner.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchsnapshot/partitioner.py b/torchsnapshot/partitioner.py index af84973..7f8efed 100644 --- a/torchsnapshot/partitioner.py +++ b/torchsnapshot/partitioner.py @@ -339,7 +339,7 @@ def consolidate_replicated_entries( if logical_path in replicated_entries: if replicated_entries[logical_path] != entry: raise ValueError( - f"Paths for replicated entry do not match: replicated entries={replicated_entries[logical_path]} vs. entry={entry}" + f"Paths for replicated entry for {logical_path} do not match: replicated entries={replicated_entries[logical_path]} vs. entry={entry}" ) else: replicated_entries[logical_path] = entry