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

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

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

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

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

相关推荐
لا معنى له3 分钟前
学习笔记:少样本学习
人工智能·笔记·深度学习·学习·机器学习
am心17 分钟前
学习笔记-菜品接口-删除菜品
笔记·学习
لا معنى له19 分钟前
学习笔记:目标跟踪内涵、方法及经典模型
人工智能·笔记·学习·计算机视觉·目标跟踪
何以不说话23 分钟前
记录一下学习日常⑨(ansible、Open-V、zabbix)
学习·ansible·zabbix
saoys25 分钟前
Opencv 学习笔记:文字图像预处理(二值化 + 闭运算优化)
笔记·opencv·学习
wdfk_prog28 分钟前
[Linux]学习笔记系列 -- compiler
linux·笔记·学习
星火开发设计43 分钟前
从公式到应用:卷积公式全面解析与实战指南
学习·算法·机器学习·概率论·知识·期末考试·卷积公式
九成宫1 小时前
计算机网络期末复习——第4章:网络层 Part Two
网络·笔记·计算机网络·软件工程
实战项目1 小时前
云原生中间件的消息队列性能优化
学习
玄〤1 小时前
黑马点评中的分布式锁设计与实现(Redis + Redisson)
java·数据库·redis·笔记·分布式·后端