Skip to content

Commit

Permalink
Remove dist_ prefix from TORCH_LOGS shortcuts (#126499)
Browse files Browse the repository at this point in the history
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: #126499
Approved by: https://github.com/XilunWu, https://github.com/kwen2501
ghstack dependencies: #126322
  • Loading branch information
wconstab authored and ZelboK committed May 19, 2024
1 parent 38a85b2 commit 8ab08f9
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions torch/_logging/_registrations.py
Original file line number Diff line number Diff line change
Expand Up @@ -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])

Expand Down

0 comments on commit 8ab08f9

Please sign in to comment.