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

LLVM MLIR怎么在mlir-opt中调用自定义Pass

冬晨小哥_9097

冬晨小哥_9097

发布时间:2026-09-29 15:46:28

|

708人浏览过

|

来源于php中文网

原创

mlir-opt不支持动态加载自定义Pass,必须重新编译并静态注册方言和Pass;正确做法是构建树外工具tutorial-opt,在Main.cpp中调用registerMyDialects()和registerMyPasses(),并确保CMake正确链接依赖库。

llvm mlir怎么在mlir-opt中调用自定义pass

不能直接用 mlir-opt 加载自定义 Pass,除非你重新编译它并注册你的方言和 Pass。 它不是 opt 那种支持 -load 动态库的工具。默认构建的 mlir-opt 只认 LLVM 项目里已注册的方言(如 std、affine、arith)和内置 Pass(如 --canonicalize),你写的任何新 Pass 都不会出现在它的命令行选项里,也不会被 --pass-pipeline 解析到。

为什么 mlir-opt 不支持 -load 或动态加载

因为 mlir-opt 的设计目标是“轻量 CLI 前端 + 静态注册”,所有方言和 Pass 必须在编译期通过 registerDialects() 和 registerPasses() 注入。它没有运行时符号解析机制,也不链接 dlopen 相关逻辑。这和 LLVM 的 opt 工具完全不同——后者明确支持 -load 加载 .so 并调用 initialize* 系列函数。

  • mlir-opt 启动时只扫描已注册的 PassRegistry 条目,没注册 = 不可见
  • 即使你把 Pass 编译进共享库,mlir-opt --help 里也绝不会出现你的 Pass 名字
  • 尝试 mlir-opt --my-custom-pass 会报错:unknown argument '--my-custom-pass'

正确做法:构建自己的 tutorial-opt

官方推荐路径是创建一个树外(out-of-tree)二进制,名字可以叫 tutorial-opt,它是 mlir-opt 的“克隆体”,但允许你插入自己的方言和 Pass。核心就三步:

  • 新建一个 tools/tutorial-opt/ 目录,放 Main.cpp(复制 mlir/tools/mlir-opt/MlirOptMain.cpp 内容)
  • 在 CMakeLists.txt 中链接你的方言库(如 add_mlir_library(MyDialect))和 Pass 库(如 add_mlir_library(MyPass))
  • 在 Main.cpp 的 main() 入口里显式调用 registerMyDialects() 和 registerMyPasses()

这样编译出的 tutorial-opt 就能识别你的 Pass,比如:tutorial-opt --my-canonicalize input.mlir。

Pass 注册必须匹配方言依赖

如果你的 Pass 操作的是自定义方言(比如 mydialect.foo_op),那它注册时必须确保该方言已注册,否则 runOnOperation() 里调用 getOperation()->getParentOfType<MyDialect>() 会失败,或者 pattern 匹配永远不触发。

  • 注册顺序很重要:先 registerMyDialects(),再 registerMyPasses()
  • Pass 类里不要硬编码方言名字符串,而要用 MyDialect::getDialectNamespace() 获取
  • 如果 Pass 依赖 affine 或 scf,也要在 CMakeLists.txt 中显式添加 MLIRAffine、MLIRSCF 链接项

调试时最容易忽略的点

Pass 在 tutorial-opt 里不生效?90% 是以下两个原因:

  • getOperation() 返回的顶层 Operation 不是你期望的 dialect scope —— 比如你写了 OperationPass<func::FuncOp>,但输入 IR 根本没 func.func,而是裸的 module,那就根本不会进入 runOnOperation
  • CMake 没把 Pass 所在的 add_mlir_library() 正确加入 tutorial-opt 的 target_link_libraries,导致符号未定义,但链接器不报错(因为 Pass 是弱注册,只是没进 registry)

验证是否注册成功:运行 tutorial-opt --help | grep my,看到你的 Pass 名字;再加 --dry-run 看是否报 “no operation matched” 还是直接跳过——前者说明注册了但没匹配上,后者说明根本没注册。

热门AI工具

更多
DeepSeek

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

豆包大模型

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

音述AI
音述AI Hot

一款AI音频处理工具,主要用于音述AI是一个以“用声音述说故事”为核心的 AI 音乐创作与声音分享社区,适合需要提升相关任务效率的用户。

Laper
Laper Hot

Laper是专为编剧、导演和制片人推出的 AI 原生剧本创作工具。

切问学术

切问学术是一款AI论文写作工具,复旦大学NLP团队推出的AI学术智能体。

Seko
Seko Hot

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

WorkBuddy

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

UpDream
UpDream Hot

一款AI视频创作工具,主要用于哔哩哔哩推出的自研AI视频创作工具,适合需要提升相关任务效率的用户。

火山引擎

火山引擎是一款面向企业的云计算与AI服务平台。

相关专题

更多
LLVM自定义Pass怎么写
LLVM自定义Pass怎么写

本专题聚焦LLVM自定义Pass开发,整理Pass类结构、run()方法、PreservedAnalyses、CMake构建、插件注册、-load-pass-plugin加载和测试用例编写流程。

0

2026.09.30

LLVM RISC-V参数配置教程
LLVM RISC-V参数配置教程

本专题介绍LLVM对RISC-V基础ISA和扩展的支持方式,涵盖RV32、RV64、标准扩展、实验性扩展、厂商扩展、-menable-experimental-extensions和版本差异。

0

2026.09.30

LLVM IR中间表示入门指南
LLVM IR中间表示入门指南

本专题整理LLVM IR的核心概念,包括中间表示作用、模块结构、函数、基本块、SSA形式、类型系统和常见语法,帮助新手理解LLVM编译流程中的关键层。

0

2026.09.30

PDF转图片方法
PDF转图片方法

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

0

2026.09.30

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

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

0

2026.09.29

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

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

200

2026.09.23

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

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

120

2026.09.23

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

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

100

2026.09.23

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

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

60

2026.09.22

热门下载

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

精品课程

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

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