log_add_exp 算子 arch35 实现当前由多个独立 DAG 节点(Max / Sub / Compare / Select / Abs / Muls / Exp / log1p 修正 / Add 等约 10 个)串联,每个节点都是一次 UB->UB 的读写,中间结果频繁在 UB 往返,成为性能瓶颈。
将整条计算链改造为寄存器内融合的 MicroAPI 自定义算子(LogAddExpSimplifiedCustom / LogAddExpFullCustom):
背景
log_add_exp 算子 arch35 实现当前由多个独立 DAG 节点(Max / Sub / Compare / Select / Abs / Muls / Exp / log1p 修正 / Add 等约 10 个)串联,每个节点都是一次 UB->UB 的读写,中间结果频繁在 UB 往返,成为性能瓶颈。
需求
将整条计算链改造为寄存器内融合的 MicroAPI 自定义算子(LogAddExpSimplifiedCustom / LogAddExpFullCustom):