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

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

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

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

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

相关推荐
灯前目力虽非昔,犹课蝇头二万言。19 分钟前
DeepSeek笔记(三):结合Flask实现以WEB方式访问本地部署的DeepSeek-R1模型
笔记·flask·deepseek
sakabu22 分钟前
ESP32 外设驱动开发指南 (ESP-IDF框架)——GPIO篇:基础配置、外部中断与PWM(LEDC模块)应用
笔记·单片机·学习·esp32
代码哈士奇23 分钟前
VitePress学习笔记
javascript·笔记·学习
sheepwjl35 分钟前
《嵌入式C语言笔记(十七):进制转换、结构体与位运算精要》
linux·c语言·开发语言·笔记·算法
optimistic_chen36 分钟前
【Java EE初阶 --- 网络原理】网络层---IP协议,数据链路层---以太网协议
网络·笔记·网络协议·tcp/ip·java-ee·文件io
小眼睛FPGA39 分钟前
【盘古100Pro+开发板实验例程】FPGA学习 | 基于紫光 FPGA 的键控 LED 流水灯
科技·学习·ai·fpga开发·fpga
天才少女爱迪生43 分钟前
pytorch的自定义 CUDA 扩展怎么学习
人工智能·pytorch·学习
YGY Webgis糕手之路1 小时前
Cesium 快速入门(三)Viewer:三维场景的“外壳”
前端·经验分享·笔记·vue·web
不可描述的两脚兽1 小时前
学习笔记《区块链技术与应用》第4天 比特币脚本语言
笔记·学习·区块链
是孑然呀2 小时前
【笔记】重学单片机(51)
笔记·单片机·嵌入式硬件