讲师中心 微信公众号
AI工具推荐 视频效率加速

LLVM MLIR怎么编写第一个优化Pass

胖萱小哥_1505

胖萱小哥_1505

发布时间:2026-09-29 15:39:08

|

735人浏览过

|

来源于php中文网

原创

LLVM和MLIR的Pass本质不同:LLVM Pass运行在llvm/目录下的传统IR(如%0 = add i32 %1, %2),使用llvm::FunctionPass;MLIR Pass运行在mlir-opt或自定义工具上,处理带方言前缀的IR(如toy.transpose),必须用mlir::OpRewritePattern或mlir::OperationPass,二者API、IR结构、构建依赖完全隔离。

llvm mlir怎么编写第一个优化pass

LLVM 和 MLIR 的 Pass 不是一个东西,别混着编

你用 opt 或 llc 跑的优化 pass 属于 LLVM Core(即 llvm/ 目录下的传统 IR),而 MLIR 的 pass 是另一套体系,运行在 mlir-opt 或自定义工具(如 tutorial-opt)上。两者 API、注册方式、IR 结构、甚至构建依赖都完全不同。试图把 LLVM 的 FunctionPass 拿去 MLIR 里注册,会直接报 undefined reference to 'mlir::OperationPassBase::runOnOperation()' 这类链接错误。

判断你该走哪条路,只看 IR 格式:如果 IR 里满是 %0 = add i32 %1, %2、define i32 @main() 这种,那是 LLVM IR,用 llvm::FunctionPass;如果 IR 里是 %0 = toy.transpose(%arg0)、func @foo() -> i32、带方言前缀(toy.、arith.、linalg.)的,那就是 MLIR,必须用 mlir::OpRewritePattern 或 mlir::OperationPass。

MLIR 中写第一个 Pass,优先选 OpRewritePattern

对新手最友好的入口不是直接继承 OperationPass,而是用 OpRewritePattern —— 它封装了匹配 + 重写的常见逻辑,避免手动遍历操作树、管理重写生命周期等底层细节。比如消除连续转置:transpose(transpose(x)) → x,只需关注“什么模式要匹配”和“怎么替换”,其余交给 PatternRewriter。

  • OpRewritePattern<TransposeOp> 表示这个 pattern 只匹配 toy.transpose 类型的操作
  • matchAndRewrite 里调用 op.getOperand().getDefiningOp<TransposeOp>() 查上一级定义,不为空就说明是嵌套转置
  • 替换时用 rewriter.replaceOp(op, {transposeInputOp.getOperand()}),不是手动画新 op 再删旧 op
  • 必须在 pass 构造函数里显式调用 addPattern<SimplifyRedundantTranspose>(getContext()),否则 pattern 不生效

注册 Pass 必须配自定义二进制,mlir-opt 默认不认识你的 Pass

官方 mlir-opt 只内置了 mlir 仓库自带的 passes(如 --canonicalize、--cse)。你写的 SimplifyRedundantTranspose 属于“树外(out-of-tree)”代码,必须自己搭一个可执行文件来加载它,典型名字是 tutorial-opt。

  • CMakeLists.txt 里要用 add_mlir_opt_library 或 add_mlir_tool 注册你的 pass 库
  • 主程序里得调用 registerPasses()(自动生成)、registerDialects(registry)(注册你用到的方言,比如 toy)、registerTranslations()(如有 lowering)
  • 命令行运行时,必须加 -load-pass-plugin=libYourPass.so(Linux/macOS)或指定插件路径,否则 tutorial-opt --my-pass 会报 unknown option
  • 调试时加 -debug-only=pattern-matcher 能看到哪些 pattern 被触发、为什么没匹配上

容易被忽略的 IR 验证与方言依赖

MLIR Pass 不像 LLVM Pass 那样能直接跑在任意 bitcode 上 —— 它强依赖方言(dialect)定义。如果你的 IR 用了 toy.transpose,但没在 pass 初始化时注册 ToyDialect,getDefiningOp<TransposeOp>() 就会返回空,因为 MLIR 不认识这个 op 类型。

  • 确保测试 IR 文件顶部有 module { ... },且所有 op 都属于已注册方言,否则 mlir-opt 解析阶段就失败
  • TransposeOp 必须从对应方言头文件(如 "toy/ToyOps.h")中 include,不能只 forward declare
  • 修改 pass 后务必 clean build 目录再 make,MLIR 的 TableGen 生成文件(*.inc)缓存极易导致 static_assert 失败或 symbol not found
  • 首次运行失败时,先用 mlir-opt --verify-diagnostics your.mlir 看是否 IR 本身就不合法,再查 pass 逻辑

热门AI工具

更多
二狗PPT
二狗PPT Hot

一款AI演示文稿工具,主要用于专为中式职场打造的AI PPT生成工具,适合需要提升相关任务效率的用户。

咔片AIPPT

一款在线AI演示文稿制作工具,可根据主题和内容需求辅助生成PPT结构与页面,提高演示材料制作效率。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

LibLibAI
LibLibAI Hot

一款AI视频创作工具,主要用于国内领先的AI创意平台,以海量模型、低门槛操作与“创作-分享-商业化”生态,让小白与专业创作者都能高效实现图文乃至视频创意表达,适合需要提升相关任务效率的用户。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

Seko
Seko Hot

一款AI视频创作工具,主要用于商汤科技推出的创编一体的AI短视频创作Agent,适合需要提升相关任务效率的用户。

Atoms
Atoms Hot

Atoms是一款AI智能体工具,第一支自动构建真实业务的 AI 团队。

Loomy
Loomy Hot

一款AI工具,主要用于科大讯飞发布的桌面级 AI 助理,比 OpenClaw 更易用、更安全!,适合需要提升相关任务效率的用户。

相关专题

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

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

200

2026.09.23

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

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

100

2026.09.23

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

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

80

2026.09.23

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

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

60

2026.09.22

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

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

60

2026.09.22

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

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

60

2026.09.22

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

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

60

2026.09.22

NumPy常见函数使用方法
NumPy常见函数使用方法

本专题整理 NumPy 常见函数使用方法相关教程,覆盖函数大全、参数用法、数组运算、统计聚合、排序处理、where 条件筛选、linspace 创建数列等常用场景,帮助读者快速掌握 NumPy 函数调用思路和实际数据处理技巧。

80

2026.09.22

NumPy性能优化版本更新与常见报错排查
NumPy性能优化版本更新与常见报错排查

本专题整理 NumPy 性能优化、版本更新与常见报错排查相关教程,覆盖向量化计算、广播性能、内存布局、NumPy 2.0 升级、版本兼容冲突、安装导入报错、dtype 溢出、矩阵运算异常和 broadcasting 报错修复,帮助读者系统掌握 NumPy 性能调优与问题定位方法。

100

2026.09.22

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn