LLVM MLIR怎么用memref读写内存

星墨姑娘_7622

星墨姑娘_7622

2026-10-03

179人浏览

原创

memref 是 MLIR 中带形状、布局和内存元数据的结构化内存引用类型,需配合 memref.load/store 访问,禁止绕过类型系统;静态全局 memref 可用 extract_aligned_pointer_as_index 提取指针传给外部函数。

llvm mlir怎么用memref读写内存

memref 是 MLIR 中表示「带形状和布局的内存引用」的核心类型,不是裸指针,也不是 C 风格数组——它自带维度、元素类型、对齐、步长(stride)和偏移(offset)信息。用它读写内存,本质是通过配套的 memref.load / memref.store 操作,在结构化语义下访问底层内存,而非直接操作地址。


memref.alloc + load/store 是最常见读写组合

动态分配一个 memref 后,必须用 memref.load 和 memref.store 访问其元素,不能用 llvm.ptr 或 llvm.inttoptr 绕过。否则会破坏类型安全,后续 lowering(如到 LLVM)可能失败或生成错误代码。

典型写法:

%0 = memref.alloc() : memref
%1 = arith.constant 1.0 : f32
memref.store %1, %0[0, 0] : memref
%2 = memref.load %0[0, 0] : memref

注意点:

  • memref.alloc 分配的是「描述符 + 数据」,描述符含 size/stride/aligned_ptr 等字段;实际数据内存由运行时(如 mlir-cpu-runner)管理
  • 下标 [0, 0] 必须与 memref 的维度匹配,越界不会报错但行为未定义(类似 C 数组)
  • memref.load 和 memref.store 的索引必须是 index 类型,不能是 i32 —— 需用 arith.index_cast 转换

静态全局 memref 要用 memref.get_global + extract_aligned_pointer_as_index

如果你需要把字符串或常量数组传给外部 C 函数(比如 puts),不能直接把 memref 当指针传:LLVM 不认识 memref 类型。得先提取对齐后的基地址:

memref.global "private" constant @msg : memref = dense
...
%0 = memref.get_global @msg : memref
%1 = memref.extract_aligned_pointer_as_index %0 : memref -> index
%2 = arith.index_cast %1 : index to i64
%3 = llvm.inttoptr %2 : i64 to !llvm.ptr<i8>
call @puts(%3) : (!llvm.ptr<i8>) -> ()
</i8></i8>

关键限制:

  • memref.extract_aligned_pointer_as_index 只在 memref 是「静态全局」且「无动态维度」(即 shape 全是常数)时才合法
  • 一旦用了 memref.alloc 或含 ? 维度(如 memref),这个提取操作就不可用 —— 此时必须走 lowering 到 LLVM 后的参数展开路径
  • 提取出的指针不带长度信息,调用 puts 这类函数前需确保末尾有 \0

动态 memref 传参到底层函数时,ABI 会自动展开成多个参数

当你把 memref 作为函数参数传入,并最终 lowering 到 LLVM,MLIR 不会把它打包成一个 struct;而是按 ABI 展开为 5 个独立参数(以 1D 为例):

LLVM 23.1.3
LLVM 23.1.3

LLVM 23.1.3 官方源码包,适合编译器安装、LLVM IR 学习、Clang 工具链构建、Pass 开发和跨平台编译实践。

下载
  • %allocated_ptr: 实际 malloc 返回的地址(可能未对齐)
  • %aligned_ptr: 对齐后的起始地址(真正用于读写的指针)
  • %offset: 偏移(单位:元素个数)
  • %size: 当前维度长度(? 维度在此处填运行时值)
  • %stride: 步长(通常为 1)

这意味着:你在 MLIR 里写 func.func @foo(%a: memref),lowering 后的 LLVM 函数签名类似:

define void @foo(float* %allocated_ptr, float* %aligned_ptr,
                  i64 %offset, i64 %size, i64 %stride)

所以你不需要、也不应该手动拆解 memref;MLIR 的 -finalize-memref-to-llvm pass 会自动处理。手动干预(比如自己构造指针算术)只会让 lowering 失败或产生不一致行为。


容易被忽略的坑:memref 维度和 layout 决定内存访问顺序

memref 默认是行主序(row-major),等价于 C 的 float[2][3];但你可以显式指定 layout:

%0 = memref.alloc() : memref>
%1 = memref.alloc() : memref>  // 列主序

layout 影响 memref.load %0[1, 2] 计算出的内存偏移位置。如果后续 lowering 到向量化或硬件后端(如 RISC-V Vector),layout 错误会导致数据错位、cache miss 加剧,甚至 silent corruption。

更隐蔽的是:memref 的 layout 信息在 memref.load/store 中不显式出现,全靠类型隐含;一旦类型丢失(比如经过不严谨的 cast 或 pattern rewrite),就很难 debug。

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
Tailwind CSS 响应式布局怎么写
Tailwind CSS 响应式布局怎么写

Tailwind CSS 响应式布局通常从移动端样式写起,再用 sm、md、lg 等前缀调整不同屏幕宽度下的网格、间距和导航。本专题整理断点配置、Grid 与 Flex 自适应布局及常见溢出问题,帮助你逐步完成多设备适配。

2026.11.03

0

19

Tailwind CSS 常用类名怎么记
Tailwind CSS 常用类名怎么记

Tailwind CSS 类名很多,先按布局、间距、文字、颜色和状态分组记忆,再通过官方文档检索具体写法。本专题结合类名排序、动态拼接失效、自定义实用类等问题,帮助你建立常用类名的速查方法。

2026.11.03

0

12

码上飞入口地址合集
码上飞入口地址合集

本专题汇总了码上飞(CodeFlying)的官方入口地址合集,涵盖官网主站及网页版在线使用入口,支持注册登录后直接在线体验零代码AI应用生成。同时整理了APP端下载渠道与小程序入口,助你快速访问平台,通过自然语言描述需求即可生成应用、小程序或网站。

2026.11.03

0

10

Uvicorn 报错与问题解决教程
Uvicorn 报错与问题解决教程

本专题汇总了Uvicorn服务器常见问题与报错解决的教程合集,涵盖启动失败、端口占用、模块导入错误、事件循环冲突及热重载失效等典型故障的排查思路。同时整理了多进程与Worker配置、SSL/HTTPS证书报错、跨域与代理配置等实战解决方案,助你快速定位并解决Uvicorn运行中的各类问题。

2026.11.02

20

14

Uvicorn使用教程
Uvicorn使用教程

本专题汇总了Uvicorn使用教程与启动运行指南,涵盖其作为Python ASGI服务器的定位与特点,以及命令行启动(uvicorn main:app --reload)、参数配置(host/port/workers)、程序内嵌运行与Gunicorn结合部署等核心用法。同时整理了热重载调试、日志配置与生产环境部署优化等技巧,助你快速掌握Uvicorn的运行与配置。

2026.11.02

20

17

Uvicorn安装配置教程
Uvicorn安装配置教程

本专题汇总了Uvicorn的安装配置教程与部署指南,涵盖通过pip安装Uvicorn、启动FastAPI/Starlette应用、开发模式热重载(--reload)及多进程工作模式配置等核心操作。同时整理了结合Gunicorn的生产部署方案、HTTPS与反向代理配置及性能调优技巧,助你快速完成Uvicorn服务的安装与上线部署。

2026.11.02

20

17

Vite 项目如何打包和部署
Vite 项目如何打包和部署

Vite项目打包部署专题汇集生产构建、产物预览、输出目录、资源路径、Nginx与GitHub Pages部署及常见白屏和路由404问题的已审核文章,帮助读者完成上线流程。

2026.10.28

120

16

Vite 如何搭配 Vue.js
Vite 如何搭配 Vue.js

Vite搭配Vue.js专题汇集创建Vue 3项目、配置开发环境、路径别名、Vue插件、代理、多环境构建与调试的已审核文章,帮助读者完成从初始化到日常开发的配置。

2026.10.28

100

15

Vite 项目目录怎么看
Vite 项目目录怎么看

Vite项目目录专题汇集入口文件、src源码、public静态资源、路径别名、项目根目录和构建输出目录相关的已审核文章,帮助读者理解各文件夹的用途与引用方式。

2026.10.28

300

10

热门下载

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

精品课程

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