État StableHLO

Lors de l'amorçage de StableHLO à partir de MHLO, nous avons hérité de l'implémentation de MHLO pour de nombreux éléments, y compris l'impression d'image, la vérification et l'inférence de forme. Grâce à cela, nous disposons déjà d'une couverture significative de l'opération, mais il reste encore beaucoup à faire pour vérifier l'exhaustivité des implémentations existantes et fournir de nouvelles implémentations lorsqu'il n'en existe aucune.

Ce document en ligne permet aux développeurs et aux utilisateurs de suivre la progression de divers aspects de l'opération : spécification, vérification, inférence de type, impression élégante, interprète, etc.

Comment l'utiliser ?

La progression d'une opération StableHLO, comme indiqué dans la ligne correspondante, sur un aspect particulier, comme indiqué dans la colonne correspondante, est suivie à l'aide de l'un des libellés de suivi suivants.

• Étiquettes génériques
  • yes: la mise en œuvre est complète.
  • no: il n'y a pas d'implémentation, mais travailler dessus fait partie de la feuille de route. Notez que le vérificateur ne peut jamais être associé à la mention "non", car l'ODS implémente déjà une vérification.
• Libellés personnalisés pour le vérificateur et l'inférence de type
  • yes: il existe une implémentation synchronisée avec la sémantique StableHLO.
  • yes*: il existe une implémentation synchronisée avec la sémantique XLA. La sémantique XLA étant souvent sous-documentée, nous utilisons hlo_verifier.cc et shape_inference.cc comme référence.
  • revisit: il y a une implémentation, mais elle ne relève pas de "oui " ni de"oui*", soit parce que nous n'avons pas encore effectué d'audit, soit parce que nous avons détecté et détecté des problèmes.
  • infeasible (infaisable) : il n'y a pas d'implémentation, car c'est impossible. C'est par exemple le cas, car le type de résultat d'une opération ne peut pas être déduit de ses opérandes et attributs.

État

Inférence de type pour les opérations quantifiées

La colonne Type Inference du tableau ci-dessus est destinée aux opérations non quantifiées. Pour la majorité des opérations quantifiées, il n'est pas possible de déduire le type de résultat, car les paramètres de quantification des types de résultats peuvent différer de ceux des opérandes. À l'exception de quelques cas où l'opérande et les types de résultats doivent correspondre de manière identique, ou l'opération comporte des contraintes utiles pour déduire le type de résultat. Les opérations suivantes sont listées ci-dessous : all_gather, all_to_all, case, collective_permute, compare, concatenate, constant, dynamic_slice, dynamic_update_slice, gather, get_tuple_element, if, infeed, all_to_all, all_to_all, all_to_all, all_to_all, all_to_all, all_to_all, all_to_allis_finitemapoptimization_barrieroutfeedpadrecvreducereduce_scatterreduce_windowreversescatterselect_and_scattersendslicesorttransposetupleuniform_dequantizedwhile