LLVM MLIR怎么用linalg表示矩阵乘法

落辰吖_1198

落辰吖_1198

2026-10-01

485人浏览

原创

linalg.matmul 语义清晰但太“裸”,无法控制 tiling、融合等,生产代码需用 linalg.generic 手动定义 indexing map、iterator types 和 region 计算逻辑。

llvm mlir怎么用linalg表示矩阵乘法

linalg.matmul 是最直接的方式,但实际项目里几乎没人只用它——因为太“裸”,没法控制 tiling、数据布局、融合或 bufferization。真正落地时,得从 linalg.generic 入手,再配合 lowering pass 链。

为什么不用 linalg.matmul 直接写生产代码

它语义清晰,适合快速验证,但参数固定、无法嵌入自定义计算逻辑,也不暴露迭代空间细节。比如你想把 bias 加进去、做 fused relu、或者适配非默认内存布局(如 column-major),linalg.matmul 就卡死了。

常见错误现象:linalg.matmul 生成的 IR 后续 pass(如 linalg.tiling)可能不生效,因为它的 op 没暴露 loop iterator 类型和 indexing map,优化器“看不清”怎么切块。

  • 它内部其实是 linalg.generic 的语法糖,但封装后隐藏了 region 和映射关系
  • 不支持动态形状下的 affine constraint 推导(比如 batch 维度 runtime unknown)
  • 在 IREE 或 TOSA 转换流程中,常被提前 lower 成 linalg.generic,你写的没意义

linalg.generic 怎么手动写矩阵乘法

核心是三件事:定义 indexing map(描述 A、B、C 各自的访存模式)、指定 iterator types(parallel/reduction)、写 region 里的 scalar 计算逻辑。

示例(A[4x8] × B[8x4] → C[4x4]):

造次
造次

一款AI视频创作工具,主要用于Liblib打造的AI原创IP视频创作社区,适合需要提升相关任务效率的用户。

下载
%0 = linalg.generic {
  indexing_maps = [
    affine_map (d0, d1)>,     // A[i, k]
    affine_map (d1, d2)>,     // B[k, j]
    affine_map (d0, d2)>      // C[i, j]
  ],
  iterator_types = ["parallel", "parallel", "reduction"],
  doc = "A * B"
} ins(%A, %B : tensor, tensor)
   outs(%C : tensor) {
  ^bb0(%a: f32, %b: f32, %c: f32):
    %mul = arith.mulf %a, %b : f32
    %add = arith.addf %c, %mul : f32
    linalg.yield %add : f32
} -> tensor

注意点:

  • 三个 indexing map 必须维度对齐:输入两个 map 输出维度都是 3,第三个 map 是输出 shape 的索引投影
  • iterator_types 顺序必须和 indexing_maps 中每个 map 的 domain 维度数一致(这里都是 3D)
  • region 内必须用 linalg.yield,且类型要和 outs 的元素类型匹配

容易踩的坑:shape、layout 和 lowering 顺序

你以为写完 linalg.generic 就能跑了?大概率会卡在 bufferization 或 vectorization 阶段。

典型报错:failed to bufferize operation 'linalg.generic': cannot bufferize a generic op with non-identity output indexing map

  • 问题出在第三个 indexing map 不是 identity —— 比如你写了 affine_map (d1, d0)>(转置输出),bufferization 默认不支持,得加 --linalg-bufferize=allow-returning-new-buffers
  • tensor 默认是 row-major,但 GPU 上常用 col-major 布局,得先用 tensor.cast 或 linalg.transpose 预处理
  • lowering 顺序不能乱:必须先 linalg.tiling(带 tile size),再 linalg.fuse(如果要融合 bias),最后 linalg.bufferize → convert-linalg-to-loops → convert-scf-to-cf

实际调试建议:用 mlir-opt 分步看 IR 变化

别一上来就跑 end-to-end pipeline。用下面命令逐层观察:

  • mlir-opt --linalg-tile='tile-sizes=4,4,8' input.mlir 看 tiling 后的嵌套 loop 结构
  • mlir-opt --linalg-bufferize --convert-linalg-to-loops input.mlir 看是否成功转成 scf.for
  • 加 --verify-diagnostics 能让报错定位到具体 op 行号,比 silent fail 强得多

最关键的细节往往藏在 indexing map 的括号里:少一个 d、多一个逗号、map 维度不一致,整个 lowering 就静默失败——它不会报错,只是跳过该 op。

相关文章

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

0

10

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

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

2026.09.30

0

14

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

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

2026.09.30

0

12

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

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

2026.09.30

0

26

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

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

2026.09.29

0

15

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

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

2026.09.23

220

15

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

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

2026.09.23

120

15

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

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

2026.09.23

100

15

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

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

2026.09.22

60

12

热门下载

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

精品课程

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