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

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

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

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

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

相关推荐
大苏打seven11 分钟前
Java学习笔记(多线程):ConditionObject 源码分析
java·笔记·学习
superior tigre37 分钟前
C++学习:六个月从基础到就业——面向对象编程:重载运算符(下)
c++·学习
汇能感知38 分钟前
光谱相机的成像方式
经验分享·笔记·科技
谢m鑫天天揍我1 小时前
【4.1.-4.20学习周报】
学习
啾啾Fun1 小时前
数据结构与算法学习导航
学习·算法
海棠蚀omo1 小时前
C++笔记-list
开发语言·c++·笔记
沙子可可1 小时前
深入学习OpenCV:第一章简介
人工智能·opencv·学习
烟雨柳成烟1 小时前
C++学习Day0:c++简介
开发语言·c++·qt·学习
stockmasterx1 小时前
什么是ETF跟踪误差?场内基金佣金最低是多少?
经验分享·笔记·其他
zhuyixiangyyds2 小时前
day30图像处理OpenCV
图像处理·笔记·学习