From 8ab08f90ef45f8b38e13be71fa135ac702263a0d Mon Sep 17 00:00:00 2001 From: Will Constable Date: Thu, 16 May 2024 18:19:09 -0700 Subject: [PATCH] Remove dist_ prefix from TORCH_LOGS shortcuts (#126499) e.g. dist_ddp -> ddp 'distributed' shortcut remains unchained Feedback has been that it is not appealing to have the dist_ prefix, and the main reason for it was to keep the distributed shortcuts grouped together in the help menu. It's nice to have shorter shortcuts. Pull Request resolved: https://github.com/pytorch/pytorch/pull/126499 Approved by: https://github.com/XilunWu, https://github.com/kwen2501 ghstack dependencies: #126322 --- torch/_logging/_registrations.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/torch/_logging/_registrations.py b/torch/_logging/_registrations.py index d76b5610e97e3..10463b864f440 100644 --- a/torch/_logging/_registrations.py +++ b/torch/_logging/_registrations.py @@ -26,13 +26,13 @@ register_log("torch", "torch") register_log("distributed", DISTRIBUTED) register_log( - "dist_c10d", ["torch.distributed.distributed_c10d", "torch.distributed.rendezvous"] + "c10d", ["torch.distributed.distributed_c10d", "torch.distributed.rendezvous"] ) register_log( - "dist_ddp", ["torch.nn.parallel.distributed", "torch._dynamo.backends.distributed"] + "ddp", ["torch.nn.parallel.distributed", "torch._dynamo.backends.distributed"] ) -register_log("dist_pp", ["torch.distributed.pipelining"]) -register_log("dist_fsdp", ["torch.distributed.fsdp"]) +register_log("pp", ["torch.distributed.pipelining"]) +register_log("fsdp", ["torch.distributed.fsdp"]) register_log("onnx", "torch.onnx") register_log("export", ["torch._dynamo", "torch.export", *DYNAMIC])