LLVM MLIR怎么为自定义操作编写校验逻辑

老辰同学_9167

老辰同学_9167

2026-10-05

693人浏览

原创

verify 函数是 mlir 中 op 语义校验的唯一自动调用入口,需在 op 类中定义签名 logicalresult verify() 并在 ods 中设 hasverifier=1;校验时可安全访问 operand、result、attr,须用 emitoperror() 报错,禁止修改 ir 或抛异常。

llvm mlir怎么为自定义操作编写校验逻辑

怎么在 MLIR 中为自定义 Op 添加校验逻辑

MLIR 不会在解析或构建时自动检查 Op 的语义合法性,verify 函数是唯一由框架主动调用的校验入口——它在 Operation 构建完成、插入到 IR 前被触发,失败则直接报错并中止插入。

校验逻辑必须写在 Op 类定义里,且需显式注册;不写 verify 或漏掉注册,等于完全跳过校验。

  • verify 是成员函数,签名固定为 LogicalResult verify(),返回 success() 或 failure()
  • 校验中可安全访问 getOperand(n)、getResult(n)、getAttr("xxx") 等,但不能修改 IR 结构
  • 错误信息必须用 emitOpError()(不是 emitError()),否则位置信息丢失、报错不指向具体 Op
  • 若依赖类型约束(如要求某 operand 是 tensor 类型且 rank ≥ 2),应先用 isa<rankedtensortype>()</rankedtensortype> 检查,再取 .getRank(),避免未检查就调用崩溃

为什么 verify 有时不触发

常见误解是“只要写了 verify 就一定运行”,实际取决于 Op 的构建方式和上下文:

Luma AI
Luma AI

Luma AI是一款AI 3D生成工具,AI 3D捕捉、建模和渲染。

下载
  • 通过 builder.create<myop>(...)</myop> 创建时,verify 默认启用且必调
  • 用 parseAssembly 从文本解析时,verify 也必调;但如果 parser 内部提前 return failure()(比如属性格式错),verify 根本不会执行
  • 在 Pass 中用 rewriter.replaceOpWithNewOp<myop>(...)</myop> 时,新 Op 的 verify 会触发,但旧 Op 已移除,不校验
  • 若 Op 定义里没在 ODS(.td 文件)中声明 let hasVerifier = 1;,C++ 代码生成器根本不会把 verify 注册进 dispatch 表——此时函数存在但永不调用

verify 里能做哪些事,哪些绝对不能做

校验阶段不是优化阶段,它的唯一职责是回答“这个 Op 在当前形态下是否合法”。越界操作会导致未定义行为或断言失败:

  • ✅ 可检查:operand 数量是否匹配、属性是否存在且类型正确、tensor shape 是否兼容、整数属性是否在合理范围(如 axis 不越界)
  • ✅ 可调用:其他 Op 提供的 hasTrait<...>()</...>、getType().isa<...>()</...>、getLoc() 获取源位置用于报错
  • ❌ 禁止:创建新 Operation、修改 Block、调用 rewriter、触发 dialect conversion、访问外部状态(如全局 map)
  • ❌ 禁止:在 verify 里抛 C++ 异常(MLIR 要求纯 LogicalResult 流程)、或用 llvm_unreachable 替代 return failure()

一个典型的 OrOp 校验示例

假设你为 Toy Dialect 添加了 toy.or,要求两个 operand 类型相同、都是 ranked tensor,且元素类型为 integer:

LogicalResult OrOp::verify() {
  auto lhs = getLhs();
  auto rhs = getRhs();
  if (lhs.getType() != rhs.getType())
    return emitOpError() ();
  if (!ty)
    return emitOpError() 

<p>注意:这里没检查 shape 是否 broadcastable——那是后续 canonicalization 或 lowering 阶段的事;<code>verify</code> 只管“结构上能否成立”,不管“语义上是否最优”。这点容易混淆,也是最常被忽略的边界。 </p>

相关文章

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

80

10

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

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

2026.09.30

80

14

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

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

2026.09.30

80

12

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

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

2026.09.30

60

26

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

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

2026.09.29

60

15

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

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

2026.09.23

280

15

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

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

2026.09.23

160

15

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

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

2026.09.23

140

15

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

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

2026.09.22

80

12

热门下载

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

精品课程

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