Lightning-AI/lightning-thunder

Make autocast a Transform

Aberta

#1.122 aberto em 7 de set. de 2024

Ver no GitHub
 (2 comentários) (0 reação) (0 responsável)Python (114 forks)github user discovery
enhancementgood first issue

Métricas do repositório

Stars
 (1.460 estrelas)
Métricas de merge de PR
 (Nenhuma PRs mesclada em 30d)

Description

Currently autocast (transforms/autocast.py) is carried over from the old-style transforms on callable and invoked like

jit(autocast(fn))

it would be good to make this a AutocastTransform to be invoked as

jit(fn, transforms=[Autocast()])

for better composability.

This should be vaguely doable with the transforms e.g. as explained in Thunder Sessions Episode 7 plus the bsym mapping provided by the existing mechanism, so it might be a good first or second issue.

Guia do colaborador

Direção de pesquisa
Estude a função autocast existente e a hierarquia de classes Transform no lightning thunder, depois implemente o AutocastTransform seguindo o padrão mostrado no Thunder Sessions Episódio 7.
Pilha de tecnologia
python
Domain
backend
Tipo Issue
Refatoração
DifficultyDificuldade de implementação estimada para um novo contribuidor, de 1 para alterações muito pequenas a 5 para trabalho de nível especializado.
2
Tempo estimadoUm intervalo de tempo aproximado para um colaborador experiente investigar, implementar, testar e preparar um pull request.
1-3 horas
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
Python
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
75