Apr, 2024

GeoT: 基于 GPU 上高效分段约简的图神经网络张量中心库

TL;DRGeoT 是一个切实针对 GNN 的创新张量中心库,通过高效的段缩减算法,引入新的设计原则,扩展了可用的设计空间,并在计算图内实现简易融合,确保与现代张量中心机器学习框架和编译器的兼容性。通过取得 1.80 倍的平均运算符加速和 1.68 倍的端到端加速,GeoT 标志着显著的进步。