facebookresearch/HolisticTraceAnalysis

Is it possible to categorize the kernels befween forward and backward passes

Ouverte

#206 ouverte le 29 nov. 2024

Voir sur GitHubGuide contributeur
 (1 commentaire) (0 réaction) (0 personne assignée)Python (94 forks)auto 404
help wantedneeds triagequestion

Métriques du dépôt

Stars
 (535 étoiles)
Métriques de merge PR
 (Métriques PR en attente)

Description

What is your question?

Thanks for the great library!

analyzer.get_gpu_kernel_breakdown(visualize=True) returns a dataframe with kernels statistics. Is it possible to split those between forward/backward categories?

Code

No response

What have you tried?

No response

Environment

No response

Guide contributeur

Direction de recherche
Examinez la fonctionnalité existante de décomposition du noyau GPU et explorez les méthodes pour taguer les noyaux par passe (avant/arrière) à l'aide d'autograd PyTorch ou de hooks du profileur.
Stack technique
pythonpytorch
Domaine
machine learningperformance
Type d'issue
Recherche
DifficultéDifficulté estimée pour un nouveau contributeur, de 1 pour un très petit changement à 5 pour un travail expert.
3
Temps estiméFourchette de temps approximative pour investiguer, implémenter, tester et préparer une pull request.
Une demi journée
Statut d'activitéDisponibilité apparente de l'issue : fraîche, active, ancienne, bloquée ou en attente d'un mainteneur.
Active
ClartéClarté avec laquelle l'issue explique le changement attendu, les critères d'acceptation et la prochaine étape.
Claire
Prérequis
PythonPyTorch
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
60