昇腾社区首页
中文
注册
开发者
下载

图引擎

  • 新增ES(Eager Style)极简构图特性,采用类似Torch Eager语法,通过函数直接表达连边与IR信息传递。相比通过算子原型(即IR)构建Graph的方式,代码量更少、易用性显著提升,且具备一定的防呆能力。特性详细介绍请单击Link
  • 新增基于Pattern匹配的自定义Pass改图特性,提供基于图结构的匹配与替换接口,大幅提升自定义融合Pass的开发效率。开发者只需继承GE基类并调用宏,即可将Pass注册到指定阶段。支持两类场景:
    • 通用子图融合(1:1或复杂拓扑替换):继承 PatternFusionPass 类,通过 REG_FUSION_PASS 宏注册。
    • 单节点替换(1节点替换为N节点):继承 DecomposePass 类,通过 REG_DECOMPOSE_PASS 宏注册。

    特性详细介绍请单击Link