LLVM MLIR怎么用scf.for表示循环

冬丽大大_2704

冬丽大大_2704

2026-10-04

281人浏览

原创

scf.for 是 mlir 中显式带范围、步长和归纳变量的 ssa 操作,非 c 风格语法糖;其基本结构为:scf.for %i = %lb to %ub step %step { ... },其中 %i 为只读归纳变量。

llvm mlir怎么用scf.for表示循环

scf.for 的基本结构长什么样

MLIR 里的 scf.for 不是 C 风格的 for 循环语法糖,它是一个显式带范围、步长和归纳变量的 SSA 操作。你不能写 for (int i = 0; i 这种形式;必须拆成三部分:起始值、终止值、步长,全部作为操作数传入。

最简合法形式是:

scf.for %i = %lb to %ub step %step {
  // 循环体
}

其中 %lb、%ub、%step 必须是 index 类型,且 %ub 是**不包含**的上界(即循环执行条件为 i )。

  • %i 是归纳变量(induction variable),类型自动推导为 index,只在循环体内可见
  • 循环体必须是单个 scf.for 区域(region),不能为空;哪怕只做一次迭代也要写个 scf.yield
  • 如果需要返回值(比如累加结果),得用 scf.reduce 或手动带出 yield 值,scf.for 本身不返回值

为什么 scf.for 里不能直接用 arith.constant 当边界

常见错误是这么写:

scf.for %i = %c0 to %c10 step %c1 { ... }

然后报错:operand #0 of 'scf.for' must be of type 'index' —— 因为 %c0 默认是 i64 或 i32,不是 index。MLIR 对类型非常严格,尤其在控制流中。

正确做法是显式转成 index:

NeoAgent
NeoAgent

一款AI办公效率工具,主要用于销售易推出的AI‑CRM智能体平台,适合需要提升相关任务效率的用户。

下载
%c0_idx = arith.index_cast %c0 : i64 to index
%c10_idx = arith.index_cast %c10 : i64 to index
scf.for %i = %c0_idx to %c10_idx step %c1_idx { ... }
  • 更推荐用 arith.constant 直接声明 index 类型常量:%c0 = arith.constant 0 : index
  • 步长也必须是 index,%c1 = arith.constant 1 : index 才合法
  • 运行时动态边界(比如来自 memref.dim)也得是 index 类型,否则编译器拒绝 lowering

嵌套 scf.for 怎么写才不踩多面体优化的坑

MLIR 的 affine 和 linalg 优化链(比如 tiling、fusion)依赖循环结构可分析。如果你手写 scf.for 嵌套但没对齐仿射约束,后续 pass 可能直接跳过或报 not a valid affine map。

关键点:

  • 所有边界和步长必须是**静态仿射表达式**:只能含常量、index 参数、加减乘(乘仅限常量系数),不能有除法、取模、条件分支
  • 避免用 memref.load 结果当循环上界——那会变成运行时值,破坏仿射性
  • 典型安全写法是把维度信息提前提取并 cast 成 index:%dim = memref.dim %A, %c0 : memref<?x ?xf32> -> index
  • 如果真要动态上界又想保留优化机会,改用 scf.while + 显式条件判断,但会丢失多面体分析能力

scf.for 和 linalg.generic 的边界选择冲突怎么解

当你从 linalg.generic lower 到 scf.for 时,编译器自动生成的循环边界可能和你手写的不一致,尤其在分块(tiling)后。比如 linalg 要求 K 维按 16 分块,但你的 scf.for 步长写死 %c8,就会导致访存越界或计算漏项。

根本原因在于:linalg 的迭代空间是语义化的(比如 “遍历 A 的 M×K 和 B 的 K×N”),而 scf.for 是底层结构化的。二者衔接靠 linalg.tiled_loop 或 scf.forall 中间表示过渡。

  • 别直接手写 scf.for 替代 linalg lowering;优先用 linalg.tile + linalg.fuse 生成结构化循环
  • 若必须介入,检查 lowering 后 IR 中 scf.for 的 lowerBound/upperBound 属性是否被 affine.min 包裹——那是编译器在处理不整除情形(如 size=65, tile=32 → 最后一块 size=1)
  • 手动验证每层循环的步长是否与目标硬件分块参数对齐(AVX2 是 M=2/N=32,AMX 是 M=32/N=32);错配会导致向量化失败

真正难的不是写出合法的 scf.for,而是让它既满足类型和仿射约束,又能被后续 lowering pass 识别为可优化的结构——边界怎么算、步长怎么设、哪些值该提前 materialize,这些细节一旦错位,后面所有向量化、bufferization、codegen 都会连锁失效。

相关文章

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

60

10

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

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

2026.09.30

40

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

40

15

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

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

2026.09.23

240

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

热门下载

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

精品课程

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