Status do StableHLO

Ao inicializar o StableHLO do MHLO, herdamos a implementação do MHLO de muitas coisas, incluindo shapedprinting, verificação e inferência de formas. Graças a isso, já temos uma cobertura significativa da opset, mas ainda há muito o que fazer para revisar a integridade das implementações atuais e fornecer novas implementações quando não houver nenhuma.

Esse documento dinâmico serve para que os desenvolvedores e usuários acompanhem o progresso em vários aspectos da opset: especificação, verificação, inferência de tipo, impressão avançada, intérprete etc.

Como usar

O progresso de uma op StableHLO, como mencionado na linha correspondente, em um aspecto específico, como mencionado na coluna correspondente, é rastreado usando um dos rótulos de rastreamento a seguir.

• Rótulos genéricos
  • yes: há uma implementação abrangente.
  • no: não há implementação, mas trabalhar nisso faz parte do roteiro. O verificador nunca pode ser rotulado como "não" porque o ODS já implementa alguma verificação.
• Rótulos personalizados para verificador e inferência de tipo
  • yes: há uma implementação e ela está sincronizada com a semântica do StableHLO.
  • yes*: há uma implementação e está sincronizada com a semântica do XLA. Como a semântica XLA muitas vezes é subdocumentada, estamos usando hlo_verifier.cc e shape_inference.cc como referência.
  • revisit: há uma implementação, mas não se enquadra em "sim" ou "sim*", seja porque ainda não fizemos a auditoria ou porque encontramos problemas e encontramos problemas.
  • inviável: não há implementação porque é inviável. Por exemplo, porque o tipo de resultado de uma op não pode ser inferido com base nos operandos e atributos dela.

Status

Inferência de tipos para operações quantizadas

A coluna Type Inference da tabela acima se destina a operações não quantizadas. Para a maioria das operações quantizadas, não é viável inferir o tipo de resultado porque os parâmetros de quantização dos tipos de resultado podem ser diferentes dos parâmetros dos operandos. Com exceção de alguns casos em que os tipos de operando e resultado precisam corresponder de forma idêntica ou a op tem restrições úteis para inferir o tipo de resultado, estas operações são listadas abaixo: all_gather, all_to_all, case, collective_permute, compare, concatenate, constant, dynamic_slice, dynamic_update_slice, gather, get_tuple_element, if, infeed, is_finite, map, , , , , mapoptimization_barrieroutfeedpadrecvreducereduce_scatterreduce_windowreversescatterselect_and_scattersendslicesorttransposetupleuniform_dequantizedwhile