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

LLVM MLIR怎么将自定义操作转换到LLVM方言

轻丽同学_4508

轻丽同学_4508

发布时间:2026-09-29 14:49:26

|

989人浏览过

|

来源于php中文网

原创

toy.print不会自动降到LLVM,因其未被标记为非法操作且缺少配套ConversionPattern;需显式标记非法、实现matchAndRewrite生成全局字符串和printf调用,并用applyFullConversion注册模式。

llvm mlir怎么将自定义操作转换到llvm方言

toy.print 这类自定义操作无法被 MLIR 的标准 lowering 框架自动转到 LLVM 方言,必须手动提供 ConversionPattern 实现。这是最常卡住的地方——不是框架不支持,而是它默认只处理标准 dialect(如 arith、memref、affine),对用户自定义的 toy::PrintOp 完全“视而不见”。

为什么 toy.print 不会自动降到 LLVM?

MLIR 的方言转换框架(DialectConversion)只对注册为“非法操作”的 op 触发 lowering,但前提是:你明确告诉它哪些是非法的、哪些目标 dialect 是合法的。默认情况下,toy::PrintOp 既没被标记为非法,也没有配套的 lowering pattern,所以它会原封不动留在 IR 里,直到报错:failed to legalize operation 'toy.print'。

  • ConversionTarget 必须显式标记 toy::PrintOp 为非法:target.markOpAsIllegal<:printop>()</:printop>
  • 仅标记非法还不够——没有 ConversionPattern,框架连“怎么降”都不知道
  • LLVM 方言本身不提供 printf 的封装,得你自己在 module 里声明或插入 LLVM::LLVMFuncOp

PrintOpLowering 的 matchAndRewrite 要写什么?

核心是两件事:生成字符串常量、调用 printf。不能直接 emit 字符串字面量,得先用 LLVM::GlobalOp 声明全局 C 字符串,再取其地址传给 printf。示例关键片段:

LogicalResult matchAndRewrite(toy::PrintOp op, ArrayRef<Value> operands,
                              ConversionPatternRewriter& rewriter) const override {
  Location loc = op.getLoc();
  auto module = op->getParentOfType<ModuleOp>();
  auto printfRef = getOrInsertPrintf(rewriter, module);

  // 创建 C 风格字符串:"\n\0"
  auto stringAttr = rewriter.getStringAttr("\n");
  auto global = rewriter.create<LLVM::GlobalOp>(
      loc, LLVM::LLVMPointerType::get(IntegerType::get(rewriter.getContext(), 8)),
      /*isConstant=*/true, LLVM::Linkage::Internal, "str", stringAttr);

  auto stringPtr = rewriter.create<LLVM::AddressOfOp>(loc, global);

  // 调用 printf("%s", ptr)
  rewriter.create<LLVM::CallOp>(loc, printfRef, IntegerType::get(rewriter.getContext(), 32),
                                     ArrayRef<Value>{stringPtr});
  rewriter.eraseOp(op);
  return success();
}
  • 别漏掉 rewriter.eraseOp(op),否则旧 op 还在,后续 pass 会重复处理或冲突
  • LLVM::AddressOfOp 是必须的——LLVM 方言不接受裸字符串字面量,只认指针
  • 如果要打印 tensor 内容,得额外加循环 + memref.load + 类型转换,不是一行 printf 能搞定的

怎么把 pattern 注册进 lowering 流程?

不是写了 class 就生效。必须在 pass 中显式插入 pattern 列表,并确保它和 ConversionTarget、TypeConverter 三者协同工作:

OwningRewritePatternList patterns(&getContext());
patterns.insert<PrintOpLowering>(&getContext());
// 注意:这里必须用 FullConversion,PartialConversion 会放过未处理的 op
if (failed(applyFullConversion(getOperation(), target, std::move(patterns))))
  signalPassFailure();
  • 用 applyFullConversion,不是 applyPartialConversion——后者允许残留非法 op,但最终生成 LLVM IR 时会失败
  • PrintOpLowering 构造函数里传的 toy::PrintOp::getOperationName() 必须和实际 op 名完全一致(含命名空间),大小写敏感
  • 如果你的 op 有 operand 或 result 类型(比如 tensor<f64>),还得配 TypeConverter 把 tensor 转成 memref,否则类型不匹配直接 crash
真正容易被忽略的是:LLVM 方言对 runtime 符号(如 printf)的依赖是隐式的。你生成了 call,但没保证链接时能 resolve ——JIT 执行前得手动注入 printf 符号,或者用 ExecutionEngine 绑定系统 libc。这步不做,createJIT 会静默失败或 segfault。

热门AI工具

更多
DeepSeek

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

PixPix
PixPix Hot

PixPix是一款面向电商视觉生产的AI商品图生成工具。

AionClaw
AionClaw Hot

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

超级简历WonderCV

一款AI办公效率工具,主要用于免费求职简历模版下载制作,应届生职场人必备简历制作神器,适合需要提升相关任务效率的用户。

墨刀AI
墨刀AI Hot

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

WorkBuddy

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

二狗PPT
二狗PPT Hot

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

豆包大模型

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

UP简历
UP简历 Hot

一款AI办公效率工具,主要用于基于AI技术的免费在线简历制作工具,适合需要提升相关任务效率的用户。

相关专题

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

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

200

2026.09.23

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

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

100

2026.09.23

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

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

80

2026.09.23

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

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

60

2026.09.22

Conan二进制包配置指南
Conan二进制包配置指南

本专题介绍Conan根据操作系统、编译器、架构和构建类型生成二进制包的方法,讲解Profile、Settings、Options及Package ID的作用,帮助管理不同平台和编译环境下的包版本。

60

2026.09.22

Conan私有仓库搭建教程
Conan私有仓库搭建教程

本专题系统的讲解Conan私有仓库的搭建流程,涵盖仓库服务部署、存储目录配置、用户认证、权限划分和远程地址添加,并介绍内部C++依赖包的上传、下载及版本维护方法。

60

2026.09.22

loomy官网入口地址合集
loomy官网入口地址合集

本专题汇总了 Loomy 桌面 AI 助理的官方入口地址合集及使用指南。提供 macOS 与 Windows 客户端下载 。Loomy 是讯飞推出的桌面级 AI 工作搭子,支持文件整理、数据分析、网页操作及通过飞书/钉钉远程操控电脑,助你高效完成本地办公任务 。

60

2026.09.22

NumPy常见函数使用方法
NumPy常见函数使用方法

本专题整理 NumPy 常见函数使用方法相关教程,覆盖函数大全、参数用法、数组运算、统计聚合、排序处理、where 条件筛选、linspace 创建数列等常用场景,帮助读者快速掌握 NumPy 函数调用思路和实际数据处理技巧。

80

2026.09.22

NumPy性能优化版本更新与常见报错排查
NumPy性能优化版本更新与常见报错排查

本专题整理 NumPy 性能优化、版本更新与常见报错排查相关教程,覆盖向量化计算、广播性能、内存布局、NumPy 2.0 升级、版本兼容冲突、安装导入报错、dtype 溢出、矩阵运算异常和 broadcasting 报错修复,帮助读者系统掌握 NumPy 性能调优与问题定位方法。

100

2026.09.22

热门下载

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

精品课程

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

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