facebookresearch/HolisticTraceAnalysis

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

Aberta

#206 aberto em 29 de nov. de 2024

Ver no GitHubGuia do colaborador
 (1 comentário) (0 reação) (0 responsável)Python (94 forks)auto 404
help wantedneeds triagequestion

Métricas do repositório

Stars
 (535 estrelas)
Métricas de merge de PR
 (Métricas PR pendentes)

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

Guia do colaborador

Direção de pesquisa
Investigue a funcionalidade existente de detalhamento do kernel da GPU e explore métodos para marcar kernels por passagem (forward/backward) usando autograd do PyTorch ou hooks do profiler.
Pilha de tecnologia
pythonpytorch
Domain
machine learningperformance
Tipo Issue
Pesquisa
DifficultyDificuldade de implementação estimada para um novo contribuidor, de 1 para alterações muito pequenas a 5 para trabalho de nível especializado.
3
Tempo estimadoUm intervalo de tempo aproximado para um colaborador experiente investigar, implementar, testar e preparar um pull request.
Meio dia
Status da atividadeQuão disponível o issue aparece agora: novo, ativo, obsoleto, bloqueado ou aguardando entrada do mantenedor.
Ativo
ClarityCom que clareza o issue explica a mudança esperada, os critérios de aceitação e a próxima etapa.
Claro
Prerequisites
PythonPyTorch
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
60