llvm/torch-mlir

Add support for `AtenDivTensorOp` `AtenDivScalarOp` with integer operands

Aberta

#456 aberto em 2 de dez. de 2021

Ver no GitHub
 (1 comentário) (0 reação) (0 responsável)C++ (724 forks)github user discovery
help wanted

Métricas do repositório

Stars
 (1.888 estrelas)
Métricas de merge de PR
 (Mesclagem média 11d 19h) (18 fundiu PRs em 30d)

Description

The type promotion rule for div ops are different when both operands are integers (integer operands are promoted to default float type) see https://pytorch.org/docs/stable/generated/torch.div.html.

To support this, AtenDivTensorOp and AtenDivScalarOp need to have some customized logic in RefineTypes rather than using visitBinaryBroadcastingOp or visitBinaryTensorScalarOp directly as it is now https://github.com/llvm/mlir-npcomp/blob/5475d02a0d03b2bda7ef9cc240773041d22a926b/lib/Dialect/Torch/Transforms/RefineTypes.cpp#L305-L308.

Once the changes in RefineTypes are made, the TorchToLinalg lowering code can be updated as well.

Guia do colaborador

Direção de pesquisa
Estude as regras de promoção de tipo para divisão no PyTorch e entenda como o RefineTypes funciona no dialeto Torch. Em seguida, implemente a lógica personalizada para AtenDivTensorOp e AtenDivScalarOp no RefineTypes.cpp e atualize o lowering TorchToLinalg.
Pilha de tecnologia
cpp
Domain
compilersmachine learning
Tipo Issue
Funcionalidade
DifficultyDificuldade de implementação estimada para um novo contribuidor, de 1 para alterações muito pequenas a 5 para trabalho de nível especializado.
4
Tempo estimadoUm intervalo de tempo aproximado para um colaborador experiente investigar, implementar, testar e preparar um pull request.
1-2 dias
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
C++MLIRPyTorch
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
35