所有课程

机器学习编译优化

章节 1: 深度学习编译器的结构

编译流程概览

计算图和有向无环图

ML 中的静态单赋值

高级IR与低级IR

动手实践：检查 Relay IR

章节 2: 图级转换

算子融合策略

死代码消除与公共子表达式消除

动手实践：实现融合遍

章节 3: 内核优化与循环调度

循环优化空间

循环分块与块化

向量化与SIMD

循环重排与展开

动手实践：优化矩阵乘法

章节 4: MLIR 架构

MLIR 架构和方言

仿射方言与分析

模式重写与降级

动手实践：创建自定义方言

章节 5: 硬件感知代码生成

GPU 内存层次结构映射

线程绑定与Warp分化

张量核内联函数

共享内存体位与冲突

动手实践：编写 Triton 核函数

章节 6: 自动调优与搜索空间

定义搜索空间

搜索算法：从随机到遗传

机器学习成本模型

Ansor 和 AutoTVM 架构

动手实践：ResNet块的自动调优

GPU 内存层次结构映射

这部分内容有帮助吗？

© 2025 ApX Machine Learning用心打造