Shortcuts

FXE0015:fx-node-insert-type-promotion

此诊断监控节点级别的类型提升插入过程。在 PyTorch 中,存在一个称为隐式类型提升的自动过程,其中操作符的输入类型被提升为通用类型。通用类型的确定基于每个操作符特定的类型提升规则。要了解更多关于 PyTorch 的类型提升规则,请参阅 elementwise_dtypes 文档torch._refs 操作

然而,ONNX 不支持隐式类型提升。因此,为了复制 PyTorch 的行为,我们需要显式插入类型转换节点。 此诊断跟踪节点级别类型提升插入的过程。

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

优云智算