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

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

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

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

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

相关推荐
啥也不会的菜鸟·5 分钟前
Redis7——进阶篇(三)
redis·学习·缓存·redis经典面试题
whennl6 分钟前
IO学习day3
学习
codexu_46122918744 分钟前
Tauri跨端笔记实战(4) - 如何实现系统级截图
前端·笔记·rust·app·tauri
Q一件事1 小时前
生态安全相关文献推荐
学习
汇能感知1 小时前
不同类型光谱相机的技术差异比较
经验分享·笔记·科技
朝九晚五ฺ2 小时前
【Linux探索学习】第三十二弹——生产消费模型:基于阻塞队列和基于环形队列的两种主要的实现方法
linux·运维·学习
老哥不老2 小时前
深入 Vue.js 组件开发:从基础到实践
vue.js·笔记
小馒头学python2 小时前
【AIGC实战】蓝耘元生代部署通义万相2.1文生图,结尾附上提示词合集
python·学习·算法·aigc
Suckerbin2 小时前
Raven: 2靶场渗透测试
数据库·学习·安全·网络安全
zzzkk20092 小时前
BambuStudio学习笔记:MeshBoolean类
笔记·学习