FXE0015: 节点插入类型提升

此诊断工具监控节点级别的类型提升过程。在 PyTorch 中,存在一个称为隐式类型提升的自动机制,在该过程中,操作符的输入类型会被提升到一个公共类型。具体公共类型的确定依据每个操作符特定的类型提升规则。要了解更多关于 PyTorch 的类型提升规则,请参阅 elementwise_dtypes 文档torch._refs 操作符

然而,ONNX 不支持隐式的类型提升。因此,为了再现 PyTorch 的行为,我们需要显式地插入转换节点。此诊断记录了在节点级别插入类型提升的进程。

此过程使用的类型提升规则可以在 torch/onnx/_internal/fx/passes/type_promotion.py 中找到。若需更新或新增类型提升规则,请参考“注意:更新类型提升规则”部分。

本页目录