跳到正文
原文
arXiv · Machine Learning Theory· Robert Hu·· 5 小时前AI 评分33

Format-Aware Fusion for Fast FP4 Pretraining

Format-Aware Fusion for Fast FP4 Pretraining

AI 导读

本文提出 format-aware fusion 方法,通过协同设计量化生产者与规模域、消费者布局,在 native \mxfp{}、global \nvfp{} 和 cooperative-thread-array-local \nvfp{} 上加速 FP4 预训练。

来源:arXiv · Machine Learning Theory · arxiv.org