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

秋明酱_8442

秋明酱_8442

2026-10-05

914人浏览

原创

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

示例:

灵珠AI
灵珠AI

一款AI开发辅助工具,主要用于零代码AI应用创作平台,适合需要提升相关任务效率的用户。

下载
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 第一步。

相关文章

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

40

12

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

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

2026.09.30

40

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

120

15

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

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

2026.09.22

80

12

热门下载

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

精品课程

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