LLVM MLIR怎么用OpBuilder生成第一段IR

千晨同学_6332

千晨同学_6332

2026-10-06

126人浏览

原创

必须先初始化mlircontext并加载dialect,再以moduleop为起点设置有效插入点,否则opbuilder会静默失败或崩溃;示例中两次setinsertionpoint确保constop正确插入func body。

llvm mlir怎么用opbuilder生成第一段ir

直接用 OpBuilder 生成第一段 IR,关键不是“怎么建”,而是“在哪建、以谁为上下文、插到哪”。没选对插入点或没初始化好 MLIRContext,OpBuilder 会静默失败或崩溃——这不是 bug,是设计使然。

必须先有 MLIRContext 和 ModuleOp

所有 IR 构建都依赖一个活的 MLIRContext。不能 new 一个空 context 就开干;它需要注册 dialect(比如 arith、func),否则后续 op 创建会报 unknown dialect 错误。

  • MLIRContext 必须显式启用所需 dialect:context.loadDialect<:arithdialect func::funcdialect>()</:arithdialect>
  • 推荐从 ModuleOp 入手:它自带 context,且天然提供顶层 Block —— 这是你唯一能安全调用 OpBuilder::create 的起点
  • 别试图在裸指针或未初始化的 Block* 上构造 OpBuilder;哪怕指针非空,若其 getParent() 为空或 context 不匹配,build 调用会 segfault

创建 ModuleOp 后立即获取 Block 并设插入点

ModuleOp 自带一个匿名 Block,但它的 begin() 是 end 迭代器(空 block)。你得先确保插入点有效,否则 create 什么也插不进。

  • 用 module.getBody()->push_back(newBlock) 或更稳妥地:OpBuilder builder(module.getBody()); builder.create<:funcop>(...)</:funcop>
  • 如果想插在 module 开头,用 builder.setInsertionPointToStart(module.getBody());插末尾就用 setInsertionPointToEnd
  • 错误示范:OpBuilder builder(module.getBody()->begin()) —— 此时 begin() == end(),迭代器无效,UB

用 OpBuilder 创建第一个 func + constant 示例

下面这段 C++ 代码能跑通,生成最简可执行 IR:

Viral Video
Viral Video

一款基于AI的文字转视频创作工具,可将文字创意快速转化为适合社交媒体传播的视频内容和短视频素材。

下载
auto context = std::make_unique<mlircontext>();
context->loadDialect<:arithdialect func::funcdialect builtin::builtindialect>();

OpBuilder builder(context.get());
auto module = builder.create<:moduleop>(UnknownLoc::get(context.get()));

// 切换 builder 到 module body
builder.setInsertionPointToStart(module.getBody());

// 创建 func @main()
auto funcType = FunctionType::get(context.get(), {}, {}, {});
auto func = builder.create<:funcop>(UnknownLoc::get(context.get()), "main", funcType);
func.setPrivate();

// 切换 builder 到 func body
auto *entryBlock = func.addEntryBlock(builder, UnknownLoc::get(context.get()), {}, {});
builder.setInsertionPointToStart(entryBlock);

// 插入第一个 op:arith.constant
auto constOp = builder.create<:constantop>(
    UnknownLoc::get(context.get()),
    builder.getF64FloatAttr(3.14)
);
</:constantop></:funcop></:moduleop></:arithdialect></mlircontext>

注意:这里用了两次 setInsertionPoint* —— 第一次进 module,第二次进 func body。漏掉任一环节,constOp 就不会出现在 IR 中,也不会报错,只会静默丢失。

容易被忽略的 Location 和 Type 绑定

UnknownLoc 能跑,但一旦要做 error reporting 或 debug,必须传真实 FileLineColLoc;而 arith::ConstantOp 的类型不是由值推导的,必须显式提供:builder.getF64FloatAttr(3.14) 返回的是 Attribute,op 内部靠 getType() 拿 result type,所以你得确保 attr 类型和 op 声明一致。用 IntegerAttr 却传 F64 会 assertion fail。

真正卡住新手的,往往不是语法,而是 context 没 load dialect、插入点悬空、location/type 没对齐这三处。跑通第一段 IR 后,再补 dialect 注册和 location 构造,比从一堆报错里反推要快得多。

相关文章

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

热门下载

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

精品课程

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