Skip to content

Instantly share code, notes, and snippets.

@wconstab

wconstab/gist:6b185f772251beffc9f4b05907b3efe9

Created November 18, 2024 23:13
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save wconstab/6b185f772251beffc9f4b05907b3efe9 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save wconstab/6b185f772251beffc9f4b05907b3efe9 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
gistfile1.txt
(Pdb) first_schedules = [[x for x in rank_data if x["state"] == "scheduled"][:1] for rank_data in all_entries.values()]
(Pdb) first_schedules = [x[0] for x in first_schedules if len(x) > 0]
(Pdb) data = [(x.get("profiling_name", ""), x.get("input_sizes", ""), x.get("collective_seq_id", None)) for x in first_schedules]
(Pdb) import tabulate
(Pdb) print(tabulate.tabulate(data))
--------------- ---------- --
nccl:all_reduce [[262144]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[262144]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[262144]] 39
nccl:all_reduce [[262144]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[262144]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[262144]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[262144]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[262144]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[262144]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[262144]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[262144]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[262144]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[262144]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[262144]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[262144]] 39
nccl:all_reduce [[262144]] 39
nccl:all_reduce [[262144]] 40
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[262144]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[262144]] 39
nccl:all_reduce [[262144]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
nccl:all_reduce [[1025]] 39
--------------- ---------- --
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment