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

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

落辰吖_1198

落辰吖_1198

发布时间:2026-10-01 14:24:36

|

485人浏览过

|

来源于php中文网

原创

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]):

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

注意点:

  • 三个 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。

热门AI工具

更多
豆包大模型

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

WorkBuddy

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

PixTV
PixTV Hot

PixTV是一款面向AIGC内容创作的AI视频生成工具。

立刻MV
立刻MV Hot

立刻MV是一款AI文本写作工具,AI 音乐视频(MV)创作工具。

Loomy
Loomy Hot

一款AI工具,主要用于科大讯飞发布的桌面级 AI 助理,比 OpenClaw 更易用、更安全!,适合需要提升相关任务效率的用户。

UpDream
UpDream Hot

一款AI视频创作工具,主要用于哔哩哔哩推出的自研AI视频创作工具,适合需要提升相关任务效率的用户。

DeepSeek

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

SkildArt
SkildArt Hot

SkildArt是一款AI文本写作工具,一站式 AI 视觉创作平台。

墨刀AI
墨刀AI Hot

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

相关专题

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

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

20

2026.09.30

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

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

0

2026.09.30

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

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

20

2026.09.30

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

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

20

2026.09.30

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

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

20

2026.09.29

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

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

220

2026.09.23

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

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

140

2026.09.23

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

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

120

2026.09.23

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

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

60

2026.09.22

热门下载

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

精品课程

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

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