趋近智
机器学习编译优化
章节 1: 深度学习编译器的结构
编译流程概览
计算图和有向无环图
ML 中的静态单赋值
高级IR与低级IR
动手实践：检查 Relay IR
章节 2: 图级转换
算子融合策略
代数简化
布局转换
死代码消除与公共子表达式消除
动手实践：实现融合遍
章节 3: 内核优化与循环调度
循环优化空间
循环分块与块化
向量化与SIMD
循环重排与展开
动手实践：优化矩阵乘法
章节 4: MLIR 架构
MLIR 架构和方言
Linalg 方言
仿射方言与分析
模式重写与降级
动手实践：创建自定义方言
章节 5: 硬件感知代码生成
GPU 内存层次结构映射
线程绑定与Warp分化
张量核内联函数
共享内存体位与冲突
动手实践：编写 Triton 核函数
章节 6: 自动调优与搜索空间
定义搜索空间
搜索算法：从随机到遗传
机器学习成本模型
Ansor 和 AutoTVM 架构
动手实践：ResNet块的自动调优
章节 5: 硬件感知代码生成
课程章节
5.1 GPU 内存层次结构映射
5.2 线程绑定与Warp分化
5.3 张量核内联函数
5.4 共享内存体位与冲突
5.5 动手实践：编写 Triton 核函数
© 2025 ApX Machine Learning用心打造