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

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

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

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

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

相关推荐
南境十里·墨染春水4 分钟前
linux学习进展 进程间通讯——共享内存
linux·数据库·学习
光影少年10 分钟前
中级前端需要会的东西都有那些?
前端·学习·前端框架
斯维赤27 分钟前
Python学习超简单第八弹:连接Mysql数据库
数据库·python·学习
herinspace1 小时前
如何解决管家婆辉煌零售POS中显示的原价和售价不一致?
网络·人工智能·学习·excel·语音识别·零售
_李小白1 小时前
【AI大模型学习笔记之平台篇】第六篇:安卓开发AI工具介绍(Android CLI、Android Skill和Android Knowledge Base)
人工智能·笔记·学习
YaBingSec1 小时前
玄机靶场:供应链安全-供应链应急-Part2 通关笔记
java·笔记·安全
qeen872 小时前
【算法笔记】双指针及其经典例题解析
c++·笔记·算法·双指针
斯维赤2 小时前
每天学习一个小算法:选择排序
java·学习·算法
lizhihai_992 小时前
股市学习心得-股市的一天
学习