AI 编译器学习笔记之十六 -- TVM

1、使用tvm::Dump调试打印,比如 tvm::Dump(sch->stages)

2、每个 te.Stage 在转换为 TIR 时会生成一个对应的 tir.Block(一一对应)

  • te.Stage 是 TVM 的 Tensor Expression 层(高层次算子描述)中的调度单元,用于表示计算图中的某个操作及其优化策略(如循环展开、并行化)。

  • tir.Block 是 TVM 的 TensorIR 层(低层级中间表示)中的基本调度单元,直接对应硬件执行逻辑(如循环嵌套、内存访问)

相关推荐
大溪地C10 分钟前
CSS详细学习笔记
css·笔记·学习
LVerrrr32 分钟前
Missashe考研日记-day32
学习·考研
chennalC#c.h.JA Ptho1 小时前
Centos系统详解架构详解
linux·经验分享·笔记·系统架构·系统安全
饕餮争锋1 小时前
Spring普通配置类 vs 自动配置类-笔记
java·笔记·spring
Aimyon_361 小时前
Java复习笔记-基础
java·开发语言·笔记
吃货界的硬件攻城狮2 小时前
【STM32 学习笔记】ADC数模转换器
笔记·stm32·单片机·学习
菜鸟破茧计划2 小时前
C++ 算法学习之旅:从入门到精通的秘籍
c++·学习·算法
海尔辛2 小时前
学习黑客 MAC 地址深入了解
学习·macos·php
喜欢吃燃面2 小时前
C++:扫雷游戏
c语言·c++·学习
小浪学编程2 小时前
C#学习7_面向对象:类、方法、修饰符
开发语言·学习·c#