DCC: Data-Centric Compilation of Machine Learning Kernels for Processing-In-Memory Architectures
DCC: 面向处理-内存架构的机器学习内核数据驱动编译
机构 * University of Toronto(多伦多大学) ; Vector Institute(向量研究所) ; Barcelona Supercomputing Center(巴塞罗那超级计算中心) ; ETH Zürich(苏黎世联邦理工学院) ; Nvidia(英伟达) ; Max Planck Institute for Software Systems(马克斯·普朗克软件系统研究所)
AI总结 针对PIM系统中数据重排与计算代码优化相互依赖的问题,提出数据驱动的编译器DCC,通过统一调优联合优化数据重排与计算代码,在HBM-PIM和AttAcc PIM上分别实现平均2.21倍和3.92倍的加速。