「模型编译」标签文章

为 Trainium 编译模型：TorchNeuron 的图优化实践

TorchNeuron 把 PyTorch 图编译到 Trainium 时，性能差距主要由静态 shape、算子融合和并行切分决定；文中给出 trn1.32xlarge 上的实测编译命令、编译日志判读方法与监控阈值。