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

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

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

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

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

相关推荐
虾球xz1 小时前
CppCon 2014 学习:Making C++ Code Beautiful
c++·学习
__如果3 小时前
深度学习复习笔记
人工智能·笔记·深度学习
锦瑟居士4 小时前
Docker 笔记 -- 借助AI工具强势辅助
笔记·docker·eureka
find_element_by_id5 小时前
adb 常用命令笔记
笔记·adb
清晨朝暮5 小时前
【Linux 学习计划】-- 命令行参数 | 环境变量
linux·运维·学习
jiuweiC7 小时前
hive 笔记
hive·hadoop·笔记
2301_803652747 小时前
tiktoken学习
学习
快乐肚皮11 小时前
Redisson学习专栏(三):高级特性与实战(Spring/Spring Boot 集成,响应式编程,分布式服务,性能优化)
spring boot·学习·spring·redisson·高级特性
Rousson11 小时前
硬件学习笔记--64 MCU的ARM核架构发展及特点
笔记·单片机·学习
FAREWELL0007512 小时前
Unity基础学习(十二)Unity 物理系统之范围检测
学习·unity·c#·游戏引擎·范围检测