LLVM MLIR怎么设置方言转换的合法操作

夜瑶小哥_3712

夜瑶小哥_3712

2026-10-02

632人浏览

原创

conversiontarget 的核心是判定 operation 合法性:addlegaldialect() 仅声明方言合法,不保证其所有 operation 合法;需用 addlegalop 显式放行关键 operation(如 func.return),或用 adddynamicallylegalop 基于上下文动态判断;mlir 19+ 后 standardopsdialect 已拆分,旧用法会导致失败。

llvm mlir怎么设置方言转换的合法操作

设置 ConversionTarget 的合法 dialect 和 operation

MLIR 方言转换的核心是 ConversionTarget,它决定哪些 operation 被视为“合法”——即无需被重写或替换,可直接保留在目标 IR 中。不合法的操作才会触发你注册的 OperationConversionPattern 去降格或重写。

最常见错误是只加了 dialect 却漏掉具体 operation,导致某些看似属于该 dialect 的 operation 仍被判定为非法(比如 std.return 在 StandardOpsDialect 中,但若 dialect 版本较新或启用了模块化拆分,它可能已移入 cf 或 func dialect)。

  • target.addLegalDialect<:affinedialect mlir::standardopsdialect>()</:affinedialect> 只声明 dialect 合法,不保证其所有 operation 都自动合法
  • 若需明确放行某个 operation(如 func.return),应调用 target.addLegalOp<:func::returnop>()</:func::returnop>
  • 若某 operation 仅在特定条件下合法(如仅当 operand 类型是 memref 时才允许 affine.load),要用 target.addDynamicallyLegalOp 并传入判断 lambda
  • 注意 StandardOpsDialect 已在 MLIR 19+ 中被拆分为 arith、cf、func、memref 等更细粒度 dialect;继续用旧名会导致编译失败或静默跳过

为什么 addLegalOp 比 addLegalDialect 更安全

当你写 target.addLegalDialect<:func::funcdialect>()</:func::funcdialect>,MLIR 会检查该 dialect 的 isAllowedOp 实现,而默认实现往往只放行 core op(如 func.func),却把 func.return、func.call 排除在外——因为它们现在属于 func dialect 的子集,但逻辑上被归类为“control flow 辅助操作”。

所以实际工程中更推荐显式列举:

Hedra
Hedra

Hedra是一款用于生成说话角色、口型同步和角色视频的 AI 视觉创作平台。

下载
  • target.addLegalOp<:func::funcop mlir::func::returnop mlir::func::callop>()</:func::funcop>
  • target.addLegalOp<:arith::addiop mlir::arith::constantop>()</:arith::addiop>
  • 对自定义方言,必须确保 MyOp::hasTrait<:optrait::isterminator>()</:optrait::isterminator> 等 trait 不冲突,否则 addLegalOp 可能因验证失败而静默失效

动态合法性判断:addDynamicallyLegalOp 的典型用法

有些 operation 合法性依赖上下文,比如 toy.print 在调试阶段允许存在,但在生成 LLVM IR 前必须全部消除;又或者 affine.for 仅当 loop bounds 可静态推导时才允许保留,否则必须展开或转成 scf。

这时要用 addDynamicallyLegalOp:

  • target.addDynamicallyLegalOp<toyprintop>([&](ToyPrintOp op) { return !op->getParentOfType<:llvm::llvmdialect>(); });</:llvm::llvmdialect></toyprintop> —— 表示只要不在 LLVM dialect 区域内就合法
  • target.addDynamicallyLegalOp<:affine::forop>([](mlir::affine::ForOp op) { return op.getLowerBound().getDefiningOp<:arith::constantop>() && op.getUpperBound().getDefiningOp<:arith::constantop>(); });</:arith::constantop></:arith::constantop></:affine::forop> —— 仅当上下界都是常量时才合法
  • 注意 lambda 参数是 operation 的 const ref,不能在里面调用 rewriter 或修改 IR;它只做判断,不参与重写

容易忽略的兼容性陷阱

MLIR 的 ConversionTarget 行为在不同版本间有细微差异。2025 年后主线默认启用 strict mode:未显式声明合法的 operation,即使属于已添加的 dialect,也会被标记为 illegal,并触发 “no pattern matched” 错误。

  • 升级 MLIR 后若出现大量 conversion failed: no matching pattern,优先检查是否遗漏 addLegalOp 调用
  • target.markUnknownOpDynamicallyLegal([](Operation*){return true;}) 是临时兜底手段,但会掩盖真实问题,不建议长期使用
  • 调试时可用 target.setDebugAction([](const mlir::Operation& op, mlir::ConversionTarget::Action action) { llvm::dbgs() 查看每个 operation 的判定结果

真正麻烦的从来不是写对一行 addLegalOp,而是搞清当前 pipeline 中哪些 operation 实际会被 encounter、它们归属哪个 dialect、以及那个 dialect 是否已被拆分或弃用。

相关文章

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

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

下载

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

相关专题

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

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

2026.09.30

20

10

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

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

2026.09.30

0

14

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

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

2026.09.30

0

12

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

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

2026.09.30

20

26

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

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

2026.09.29

20

15

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

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

2026.09.23

220

15

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

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

2026.09.23

140

15

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

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

2026.09.23

120

15

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

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

2026.09.22

60

12

热门下载

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

精品课程

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