frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[FR] Don't check incomplete ranks for printing (#160195)

Browse Source 
When just printing the ranks (`-j` option) we should skip the check for "incomplete ranks" since that doesn't affect the print

Pull Request resolved: https://github.com/pytorch/pytorch/pull/160195
Approved by: https://github.com/fduwjj
ghstack dependencies: #160097
This commit is contained in:
Howard Huang
2025-08-08 08:59:18 -07:00
committed by PyTorch MergeBot
parent 96f9fbe21a
commit 65053c03a3
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tools/flight_recorder/components/builder.py
View File

@ -402,15 +402,15 @@ def build_db(
)
logger.debug("built groups, memberships")
if not args.allow_incomplete_ranks:
check_no_missing_dump_files(entries, memberships)
if args.just_print_entries:
just_print_entries(
entries, _groups, _memberships, _pg_guids, args, stack_id_trace_map
)
sys.exit(0)
if not args.allow_incomplete_ranks:
check_no_missing_dump_files(entries, memberships)
tracebacks, collectives, nccl_calls = build_collectives(
entries, _groups, _memberships, _pg_guids, version
)