facebookresearch/HolisticTraceAnalysis

How to plot timeline with the label added by torch.profiler.record_function?

Offen

#99 geöffnet am 24.01.2024

Auf GitHub ansehenContributor Guide
 (2 Kommentare) (0 Reaktionen) (0 zugewiesene Personen)Python (94 Forks)auto 404
help wantedneeds triagequestion

Repository-Metriken

Stars
 (535 Sterne)
PR-Merge-Metriken
 (PR-Metriken ausstehend)

Beschreibung

What is your question?

I want to plot the timeline of training GPT with pipeline parallelism. I use torch.profiler.record_function to label some functions , such as forward_step and backward_step of each micro_batch. I want to plot these functions for all ranks into one image, how can I do it? If using the example code (https://github.com/facebookresearch/HolisticTraceAnalysis/blob/main/examples/plot_timeline.ipynb), I get an image with kernels rather than record_function.

Contributor Guide

Research-Richtung
Erkunden Sie die API von HolisticTraceAnalysis zum Filtern von Ereignissen nach Namen, insbesondere record function Label, und ändern Sie das Beispiel zur Zeitachsen Darstellung, um diese Ereignisse einzuschließen.
Tech Stack
pythonpytorch
Domain
performancemachine learning
Issue Type
Recherche
SchwierigkeitGeschätzte Implementierungsschwierigkeit für neue Contributors, von 1 für sehr kleine Änderungen bis 5 für Expertenarbeit.
2
Geschätzte ZeitEin grober Zeitrahmen, um zu recherchieren, zu implementieren, zu testen und einen Pull Request vorzubereiten.
1-3 Stunden
AktivitätsstatusWie verfügbar das Issue gerade wirkt: frisch, aktiv, stale, blockiert oder wartet auf Maintainer-Input.
Aktiv
KlarheitWie klar das Issue die erwartete Änderung, Akzeptanzkriterien und den nächsten Schritt erklärt.
Klar
Voraussetzungen
PythonPyTorch
EinsteigerfreundlichkeitEin Score von 1 bis 100, der schätzt, wie zugänglich dieses Issue für First-Time Contributors ist.
65