llvm/torch-mlir

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

Open

#456 aberto em 2 de dez. de 2021

Ver no GitHub
 (1 comment) (0 reactions) (0 assignees)C++ (692 forks)github user discovery
help wanted

Métricas do repositório

Stars
 (1.840 stars)
Métricas de merge de PR
 (Métricas PR pendentes)

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