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

LLVM MLIR怎么用func方言定义和调用函数

秋明酱_8442

秋明酱_8442

发布时间:2026-10-05 14:54:10

|

914人浏览过

|

来源于php中文网

原创

MLIR中函数必须用func.func定义,显式声明类型、返回值并以func.return终止;调用需用func.call且参数类型严格匹配;func.return不可省略或替换,旧std方言已废弃。

llvm mlir怎么用func方言定义和调用函数

func方言里怎么定义函数

MLIR 的 func 方言(现为 func dialect,旧版叫 std)是定义和组织函数的标准方式。它要求显式声明参数类型、返回类型、函数体块,并用 func.return 终止。不支持隐式返回或省略类型。

常见错误现象:写成 func @foo() { ... } 缺少类型签名 → 报错 expected function type;或漏掉 func.return → IR 验证失败,提示 block not terminated。

  • func.func 是定义函数的操作名,不是关键字,必须带完整命名空间(func.func,不是 func 或 function)
  • 参数和返回类型必须写在括号内,格式为 (%arg0 : i32, %arg1 : f32) -> (i32),即使单返回也要用括号
  • 函数体必须是一个 Region,里面至少有一个 Block,且该 Block 必须以 func.return 或 func.call 等 terminator 操作结尾
  • 函数名以 @ 开头,如 @add;变量名以 % 开头,如 %result

示例:

func.func @add(%a : i32, %b : i32) -> i32 {
  %sum = arith.addi %a, %b : i32
  func.return %sum : i32
}

func方言里怎么调用函数

调用必须用 func.call,不能用 call 或裸函数名。调用前需确保被调函数已在模块中定义或声明(func.func 或 func.func external),否则 mlir-opt 会报 use of undefined symbol。

容易踩的坑:传参个数/类型不匹配不报编译错误,但后续 lowering(比如到 LLVM)时会崩溃或生成非法 IR;另外,func.call 的结果必须绑定到新值名(%ret),不能丢弃。

  • func.call 的第一个 operand 是符号引用(@add),不是字符串字面量,也不是 "add"
  • 参数列表必须与目标函数签名完全一致,包括顺序、数量、类型;i32 和 index 不可混用
  • 如果被调函数有多个返回值,func.call 返回一个元组值,需用结构化绑定或 std.tuple_extract(取决于方言版本)
  • 跨模块调用需用 func.func external 声明,且符号名必须全局唯一

示例:

func.func @main() -> i32 {
  %x = arith.constant 2 : i32
  %y = arith.constant 3 : i32
  %z = func.call @add(%x, %y) : (i32, i32) -> i32
  func.return %z : i32
}

为什么 func.return 不能省略,也不能用其他 terminator 替代

func.return 是 func 方言强制要求的 terminator,它携带返回值语义并参与类型检查。用 std.return(旧 std dialect)、cf.return 或直接 return 都会触发验证失败,因为这些操作不属于 func 方言,类型系统无法推导控制流出口与函数签名的匹配关系。

性能影响:无额外开销,func.return 在 lowering 到 LLVM 时直接映射为 ret 指令;但若误用非 func terminator,会导致 mlir-opt --convert-func-to-llvm 等 pass 跳过该函数或中途 abort。

  • LLVM lowering 要求所有 func.func 函数块以 func.return 结尾,否则转换失败并报 failed to legalize operation 'func.return' 类似错误
  • 如果你在写自定义 Pass,访问函数返回值必须通过 func::ReturnOp,而非泛化的 Operation::hasTrait<:isterminator>()</:isterminator>
  • 没有 void 返回:即使函数不返回值,也得写 -> () 并用 func.return,不能省略返回部分

func方言和旧 std dialect 的兼容性陷阱

MLIR 16.0+ 已将函数相关操作全迁入 func dialect,std.func、std.return 等已废弃。但很多文档、示例甚至部分工具链(如某些 mlir-translate 版本)仍默认启用 std,导致混合使用时 silently 失败。

典型症状:func.func 定义能 parse,但 func.call 报 unknown operation 'func.call';或 mlir-opt --convert-std-to-llvm 成功,而 --convert-func-to-llvm 报未注册 dialect。

  • 确认 dialect 启用:命令行加 --load-pass-plugin=libMLIRFuncTransforms.so(若手动构建)或确保 build 时启用了 MLIRFuncDialect
  • CMake 构建 LLVM/MLIR 时必须包含 -DLLVM_ENABLE_PROJECTS=mlir,且不能漏掉 MLIRFuncDialect 库(通常自动启用)
  • 检查 IR 是否合法:用 mlir-opt --verify-diagnostics input.mlir,它会明确指出哪个操作属于未注册 dialect
  • 避免混用:不要在一个 .mlir 文件里同时出现 func.func 和 std.func;迁移旧代码时,把所有 std. 前缀换成 func.,并更新 terminator 和调用语法

真正容易被忽略的是:函数签名中的类型必须是 func dialect 兼容的底层类型(如 i32、f64、tensor),不能是方言专属类型(如 torch.vtensor)——除非你已注册对应 conversion pattern。否则 func.call 会因类型不可表示而卡在 lowering 第一步。

热门AI工具

更多
WorkBuddy

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

讯飞绘文

讯飞绘文是一款由科大讯飞推出的一站式 AIGC 内容运营平台。

Atoms
Atoms Hot

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

AionClaw
AionClaw Hot

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

讯飞智作

讯飞智作是一款AI视频创作工具,AI文本配音工具,数字人课程、营销视频制作。

Laper
Laper Hot

Laper是专为编剧、导演和制片人推出的 AI 原生剧本创作工具。

DeepSeek

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

豆包大模型

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

蛙蛙写作

一款AI论文写作工具,主要用于超级AI智能写作助手,适合需要提升相关任务效率的用户。

相关专题

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

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

80

2026.09.30

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

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

80

2026.09.30

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

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

40

2026.09.30

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

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

40

2026.09.30

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

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

60

2026.09.29

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

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

280

2026.09.23

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

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

160

2026.09.23

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

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

120

2026.09.23

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

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

80

2026.09.22

热门下载

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

精品课程

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

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