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

LLVM MLIR怎么创建一个自定义Dialect

冬宇君_6090

冬宇君_6090

发布时间:2026-10-02 08:45:31

|

136人浏览过

|

来源于php中文网

原创

最稳妥路径是用TableGen+.td文件定义Dialect:需继承Dialect基类,填全name(须与IR前缀一致)、cppNamespace(如"::toy")、summary、description四字段,并include "mlir/IR/OpBase.td";生成需用mlir-tblgen -gen-dialect-decls/-defs,且必须显式调用registerDialect<ToyDialect>()完成注册。

llvm mlir怎么创建一个自定义dialect

直接用 TableGen + .td 文件定义是最稳妥的路径,手写 C++ Dialect 类容易漏注册、错命名、不兼容后续 MLIR 版本的接口变更。

怎么写一个合法的 Dialect .td 文件

核心是继承 Dialect 基类并填好四个必填字段:命名空间、C++ 命名空间、简介、描述。注意 let name 必须和你在 MLIR IR 中写 Op 时用的前缀完全一致(比如 toy.print 对应 let name = "toy")。

常见错误现象:

  • mlir-tblgen 报错说 unknown type 'Dialect':没加 include "mlir/IR/OpBase.td"
  • 生成的 C++ 编译失败,提示 getDialectNamespace 未定义:漏了 let cppNamespace,或值写成 "toy" 却没加双冒号前缀(应为 "::toy")
  • 运行期报 dialect 'xxx' is not registered:生成的 C++ 没在 initialize() 里调用 registerDialect<XXXDialect>()

最小可用模板:

include "mlir/IR/OpBase.td"

def Toy_Dialect : Dialect {
  let name = "toy";
  let summary = "Toy dialect for tutorial";
  let description = [{ Toy language dialect }];
  let cppNamespace = "::toy";
}

如何用 mlir-tblgen 生成 C++ 声明和定义

mlir-tblgen 是唯一受支持的生成器,不能用系统自带的 llvm-tblgen —— 后者缺少 MLIR 特定的后端。

两个命令必须都执行:

  • mlir-tblgen -gen-dialect-decls Ops.td -I $MLIR_INCLUDE_DIR > ToyDialect.h:生成头文件,含 class ToyDialect 声明和 getDialectNamespace()
  • mlir-tblgen -gen-dialect-defs Ops.td -I $MLIR_INCLUDE_DIR > ToyDialect.cpp:生成实现,含 initialize() 和注册逻辑

关键点:

  • -I 路径必须指向 mlir/include,不是 llvm/include
  • 生成的 ToyDialect.cpp 里默认不含 registerDialect<ToyDialect>() 调用——你得手动在 initialize() 函数末尾加上这行
  • 生成的头文件会依赖 mlir/IR/Dialect.h,确保你的 CMakeLists.txt 正确 link MLIRIR

为什么 registerDialect 必须显式调用

MLIR 不靠静态构造函数自动注册 Dialect,而是要求你在某个初始化入口(通常是 registerAllDialects() 或自定义初始化函数)中显式调用 registerDialect<XXXDialect>()。

否则即使编译通过,MLIRContext 也完全不知道这个 Dialect 存在,任何尝试解析 toy.print 的操作都会失败。

典型集成方式:

void registerToyDialect(mlir::DialectRegistry &registry) {
  registry.insert<toy::ToyDialect>();
}

然后在创建 MLIRContext 前传入:

mlir::DialectRegistry registry;
registerToyDialect(registry);
auto context = std::make_unique<mlir::MLIRContext>(registry);

手写 C++ Dialect 类的风险点

虽然可行,但极易出错:

  • getDialectNamespace() 返回值类型必须是 llvm::StringRef,返回局部字符串字面量(如 return "toy")会导致悬垂引用
  • initialize() 里若忘记调用 addOperations<...>() 注册 Op,后续所有 Op 解析都会失败,错误信息极不直观(常表现为 unknown op)
  • MLIR 15+ 版本开始要求 Dialect 类必须提供 materializeConstant 方法(哪怕空实现),否则 canonicalize pass 会崩溃
  • 没有 ODS 支持,Op 的 verify、parse、print 全得手写,且必须严格匹配 OperationState 内存布局,稍有偏差就触发断言或越界读

除非你正在调试 TableGen 本身,否则不要碰纯手写路径。

热门AI工具

更多
DeepSeek

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

Loomy
Loomy Hot

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

二狗PPT
二狗PPT Hot

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

WorkBuddy

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

Seko
Seko Hot

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

墨刀AI
墨刀AI Hot

一款AI图像与设计工具,主要用于产品经理的专属智能体,适合需要提升相关任务效率的用户。

Atoms
Atoms Hot

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

豆包大模型

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

AionClaw
AionClaw Hot

AionClaw是一款面向办公、创作和编程任务的AI桌面智能体。

相关专题

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

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

20

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编译流程中的关键层。

20

2026.09.30

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

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

20

2026.09.30

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

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

20

2026.09.29

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

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

220

2026.09.23

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

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

140

2026.09.23

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

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

120

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