LLVM MLIR怎么用CSE消除重复计算

星磊君_9165

星磊君_9165

2026-10-08

537人浏览

原创

mlir中cse pass通过mlir::createcsepass()调用,作用于ssa形式的operation树,仅在同一基本块内识别无副作用的等价表达式并复用首次计算结果,不跨block、不跨函数、不跨dialect,且受fastmath、别名属性及动态操作限制。

llvm mlir怎么用cse消除重复计算

MLIR里直接调用createCSEPass就行

MLIR本身不自己实现CSE逻辑,而是复用LLVM基础设施里的createCSEPass。这个Pass作用在SSA形式的Operation树上,自动识别同一基本块(Block)内语义等价、无副作用干扰的重复表达式,并把后续出现替换成第一次计算的结果变量。

常见使用方式是往PassManager里加一行:

pm.addPass(mlir::createCSEPass());

注意它只对当前作用域生效——比如加在func::FuncOp层级,就只优化该函数内部;加在Module层级,则会跨函数尝试(但实际受限于Symbol可见性,通常仍限于单个函数)。

容易踩的坑:

  • createCSEPass默认不跨Block做消除,循环体外和循环体内算两次,不算公共子表达式
  • 如果两个计算之间有写内存、调用外部函数、或任何标记为hasSideEffects的Op,CSE会主动跳过
  • 浮点运算受fastmath属性影响:没加arithmetic.fastmath时,addf x, y和addf y, x可能不被识别为等价(尽管数学上成立)

LLVM IR阶段用opt -enable-cse或-O2

LLVM IR层面的CSE由MachineCSE(针对机器码)和GVN(Global Value Numbering,更激进)共同覆盖,但最直接对应传统CSE的是-enable-cse选项。不过日常几乎不用手动开它,因为-O2及以上已默认包含。

验证是否生效,可以用:

opt -O2 -print-after=cse input.ll

观察输出中重复的%x = add i32 %a, %b是否只剩一个定义,其余全变成use %x。

Aiva.ai
Aiva.ai

Aiva.ai是一款用于生成多风格原创曲目的 AI 音乐作曲助手。

下载

关键限制:

  • LLVM CSE只作用于Function内,不跨函数
  • 对指针运算、数组索引这类带getelementptr的表达式,需确保地址计算路径完全一致,否则不合并
  • 调试信息(!dbg)不影响CSE判断,但若某条指令带!noalias或!invariant.load,可能改变别名分析结果,间接影响CSE范围

Triton这类DSL如何触发CSE

Triton在make_ttir阶段通过passes.common.add_cse(pm)插入CSE Pass,本质就是调MLIR的createCSEPass。但它加在TTIR(Triton Tensor IR)层,比原始MLIR更贴近GPU张量语义。

你写的Triton kernel里出现多次a + b,只要满足:

  • 都在同一个for循环块内(或同级Block)
  • 中间没穿插tl.store、tl.atomic_add等有副作用的操作
  • 操作数类型、精度、fastmath标志一致

编译时就会被合并。比如这段kernel:

z = x + y
w = x + y  # 这行会被优化掉,w直接复用z

但如果写成:

z = x + y
tl.store(ptr, z)
w = x + y  # 这里不会复用z,因为store引入了副作用

CSE不是万能的:哪些重复根本消不掉

有些“看起来重复”的计算,CSE天然无能为力:

  • 跨基本块的重复,比如if分支里各算一次a * b,CSE不合并(得靠GVN或LoopInvariantCodeMotion)
  • 不同Dialect的Op,比如arith.addi和linalg.generic里封装的加法,CSE不认为等价
  • 含非常量索引的tensor load:load A[i]和load A[j]即使i == j运行时成立,编译期也无法证明,故不消除
  • 涉及tensor.extract_slice等带动态尺寸的Op,shape未折叠前,CSE无法判定等价性

真正要压榨性能,得配合canonicalizer先规一化表达式形态,再让CSE进场——顺序错了,CSE看到的就不是“重复”,而是“长得像的两坨不同东西”。

相关文章

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

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

下载

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

相关专题

更多
FrankenPHP集成Laravel详细教程
FrankenPHP集成Laravel详细教程

本专题提供FrankenPHP集成Laravel的详细配置指南,全面解析运行原理、开发环境搭建、Caddyfile配置、Octane工作模式、数据库连接、队列任务、定时任务和生产环境优化,解决部署过程中常见的报错与兼容性问题。

2026.10.08

0

20

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

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

2026.09.30

120

10

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

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

2026.09.30

100

14

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

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

2026.09.30

80

12

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

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

2026.09.30

80

26

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

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

2026.09.29

100

15

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

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

2026.09.23

300

15

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

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

2026.09.23

180

15

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

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

2026.09.23

140

15

热门下载

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

精品课程

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