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

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

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

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

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

相关推荐
wanderful_11 分钟前
Javascript笔记分享-流程控制(超级超级详细!!!)
javascript·笔记·流程控制·实战案例·新手友好
ghgxm52013 分钟前
EXCEL使用VBA代码实现按条件查询数据库--简单实用
开发语言·数据仓库·笔记·excel·数据库开发
张人玉15 分钟前
C# WPF 折线图制作笔记(LiveCharts 库)
笔记·c#·wpf·折线图·linechart
我想我不够好。16 分钟前
低压实操 12.22
学习
知识分享小能手21 分钟前
Ubuntu入门学习教程,从入门到精通,Ubuntu 22.04 的 Vim 编辑器 —— 全面详解(含基础操作、高级技巧与编程实践)(5)
linux·学习·ubuntu
有谁看见我的剑了?24 分钟前
Linux ssh连接超时时间学习
linux·学习·ssh
upper202010 小时前
【炒股学习】集合竞价
学习·区块链
以孝治家行动10 小时前
线上共读传孝道 身体力行润家风——以孝治家家教中心开展线上学习
学习
小智RE0-走在路上11 小时前
Python学习笔记(8) --函数的多返回值,不同传参,匿名函数
笔记·python·学习
摇滚侠11 小时前
Redis 零基础到进阶,Redis 哨兵监控,笔记63-73
数据库·redis·笔记