LLVM MLIR的fold和canonicalization怎么配合使用

夜晨君_3708

夜晨君_3708

2026-09-30

776人浏览

原创

fold优先于canonicalization触发,二者协同实现“常量传播+结构简化”闭环:fold在ir构建、预处理及显式调用时轻量计算或消冗,canonicalize则通过重写模式进行跨op结构优化。

llvm mlir的fold和canonicalization怎么配合使用

fold 优先于 canonicalization 触发,但二者协同完成“常量传播 + 结构简化”闭环

fold 是什么,它在什么时机被调用

fold 是 Operation 自带的轻量级化简能力,用于在不改变语义前提下直接计算出结果(如 addi %c0, %c1 → %c1),或提前消去冗余操作(如 mul %x, %c0 → %c0)。它不依赖 Pattern Rewriter,也不修改 IR 结构,只返回新值或空列表。

常见触发点包括:

  • IR 构建时(如 builder.create<addiop>(...)</addiop> 内部会自动 fold)
  • Canonicalization Pass 执行前的预处理阶段
  • 某些转换 Pass(如 Dialect Conversion)中显式调用 op.fold(...)

canonicalization 做什么,为什么不能只靠 fold

canonicalize 是更重、更结构化的优化:它通过注册 getCanonicalizationPatterns 返回一组 RewritePattern,匹配并替换整个子图(DAG)。比如 transpose(transpose(%x)) → %x,这种跨多个 Op 的等价替换无法由单个 fold 完成。

关键差异:

Waifulabs
Waifulabs

Waifu Labs是一款用于生成和定制二次元角色头像的 AI 动漫肖像工具。

下载
  • fold 只能处理当前 Op 及其 immediate operands;canonicalize 可访问 users、nested regions、甚至跨 dialect 的 operand 依赖
  • fold 返回的是值(Value 或空),canonicalize 返回的是重写指令(PatternRewriter::replaceOp 等)
  • 一个 Op 的 fold 成功后,其 users 不会自动重试 fold;而 canonicalize Pass 会迭代运行直到无变化

两者配合的真实工作流

典型优化链路是:fold 先把局部常量暴露出来 → canonicalize 利用这些常量触发更大范围的 DAG 替换。例如:

%0 = arith.constant 0 : i32
%1 = arith.addi %x, %0 : i32  // fold 返回 [%x],直接替换为 %x
%2 = arith.muli %1, %y : i32  // 现在变成 muli %x, %y —— 更容易被 canonicalize 中的“乘法单位元” pattern 匹配

实际开发中要注意:

  • 不要在 fold 里做 heavy computation(如遍历 users 或创建新 op),否则拖慢 IR 构建
  • canonicalize pattern 的 matchAndRewrite 中,应先尝试 op.fold(...) 获取候选值,再决定是否重写,避免重复逻辑
  • 若某个 Op 同时有 fold 和 canonicalize 实现,fold 必须保证返回结果与 canonicalize 的最终效果一致,否则会出现 pass 迭代不收敛

容易被忽略的坑:fold 返回值的生命周期和 ownership

fold 返回的 Value 必须来自已有 IR(如 operand、constant、block argument),不能 new 一个临时 Value。否则会在后续 canonicalize 或验证中触发 use-def 链断裂或 verifier 失败。

典型错误写法:

// ❌ 错误:创建了未插入 block 的新 Value
auto *newOp = rewriter.create<:constantop>(...);
return std::make_pair(newOp->getResult(0), nullptr);

// ✅ 正确:复用已有 operand 或 constant
if (auto c = getOperand(0).getDefiningOp<:constantop>())
  return std::make_pair(c.getResult(), nullptr);
</:constantop></:constantop>

真正难调试的问题往往出在这里:fold 看似成功,但返回的 Value 没有合法 parent,导致后续 canonicalize 在 match 时 operand 不满足约束,静默跳过。

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
PDF转图片方法
PDF转图片方法

需要把 PDF 页面用于上传、预览、分享或图片归档时,PDF 转图片方法专题整理 JPG/PNG 格式选择、逐页导出、清晰度设置、批量下载和结果检查等流程,帮助用户稳定完成 PDF 图片化处理。

2026.09.30

0

26

PixTV AI视频生成与无限画布创作
PixTV AI视频生成与无限画布创作

PixTV专题整理AI视频与视觉内容创作相关功能使用教程,涵盖AI生图、视频生成、无限画布、多模型创作、素材管理、声音音乐及视频剪辑等功能,帮助用户快速掌握PixTV从创意到成片的完整制作方法。

2026.09.29

0

15

Buffalo框架数据库开发全教程
Buffalo框架数据库开发全教程

本专题围绕Buffalo框架数据库开发,讲解database.yml多环境配置、soda与fizz迁移生成回滚、模型结构体标签、增删改查与条件查询、一对多与多对多关联、数据校验、回调钩子、事务处理及原生SQL执行能力。

2026.09.23

200

15

Buffalo框架路由与请求处理实操指南
Buffalo框架路由与请求处理实操指南

本专题讲解Buffalo框架路由与请求处理机制,涵盖路由注册与分组、资源路由、Handler编写规范、Context上下文方法、参数绑定、中间件编写挂载、Session与Cookie读写、Flash消息及错误页面定制方法。

2026.09.23

120

15

Buffalo框架零基础入门教程
Buffalo框架零基础入门教程

本专题整理Buffalo框架入门内容,涵盖Go环境准备、buffalo CLI安装、新项目生成、目录结构说明、dev热加载启动、数据库连接配置与常见报错排查,帮助新手按约定优于配置的思路跑通第一个Buffalo框架应用。

2026.09.23

100

15

Conan创建软件包配方指南
Conan创建软件包配方指南

本专题介绍通过conanfile.py创建软件包的方法,讲解包名、版本、依赖和构建设置等基础信息,以及source、build、package、package_info等常用方法的作用及编写思路。

2026.09.22

60

12

Conan二进制包配置指南
Conan二进制包配置指南

本专题介绍Conan根据操作系统、编译器、架构和构建类型生成二进制包的方法,讲解Profile、Settings、Options及Package ID的作用,帮助管理不同平台和编译环境下的包版本。

2026.09.22

80

13

Conan私有仓库搭建教程
Conan私有仓库搭建教程

本专题系统的讲解Conan私有仓库的搭建流程,涵盖仓库服务部署、存储目录配置、用户认证、权限划分和远程地址添加,并介绍内部C++依赖包的上传、下载及版本维护方法。

2026.09.22

60

19

loomy官网入口地址合集
loomy官网入口地址合集

本专题汇总了 Loomy 桌面 AI 助理的官方入口地址合集及使用指南。提供 macOS 与 Windows 客户端下载 。Loomy 是讯飞推出的桌面级 AI 工作搭子,支持文件整理、数据分析、网页操作及通过飞书/钉钉远程操控电脑,助你高效完成本地办公任务 。

2026.09.22

80

19

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程