arrow
返回

LE-GEMM: A lightweight emulation-based GEMM with precision refinement on GPU

delete2025-03-01
delete0
PRE
AI
Y
Yu Zhang
陆
陆璐 (Lu Lu) *
Z
Zhanyu Yang
Z
Zhihong Liang
S
Siliang Suo
DOI:10.1016/j.sysarc.2025.103336delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Many special hardware units, such as Matrix Core and Tensor Core, have recently been designed and applied in various scientific computing scenarios. These units support tensor-level computation with different precisions on GPU. Previous studies have proposed methods for computing single-precision GEneral Matrix Multiplication (GEMM) with the half-precision matrix. However, this routine often leads to some loss of accuracy, which limits its application. This paper proposed a Lightweight Emulation-based GEMM (LE-GEMM) on GPU that includes a lightweight emulation algorithm, a thread parallelism analytic model, and an efficient multi-level pipeline implementation to accelerate the computation process without compromising the accuracy requirements. First, we propose a lightweight emulation algorithm that includes a precision transformation process and GEMM emulation calculation to achieve better computational accuracy and performance. Secondly, a thread parallel analytic model is designed to analyze and guide the selection of the optimal tiling scheme based on various computing scenarios and hardware. Thirdly, an efficient multi-level pipeline is implemented, which can maximize instruction-level parallelism and latency hiding. Several comparison experiments were conducted on two commonly used GPU platforms: AMD-platform and NVIDIA-platform. The experimental results show that the proposed method outperforms the previous approaches in terms of computational accuracy and speed.
Keyword:
GEMM
Thread parallelism analytic
Multi-level pipeline
Matrix core/Tensor core

期刊

Journal of Systems Architecture 封面图
Journal of Systems Architecture
IF:
4.1
论文数:
3.0K
被引数:
4.2K

机构

S
south china university of technology
学者数:
6.8W
论文数: 5.1W
被引数: 85
引用论文

引用论文

Asymmetric Information and Loan Spreads in Russia信息不对称与俄罗斯贷款利差
err2014-12-08
err0
PREAI
errZuzana Fungáčová; Christophe J. Godlewski; Laurent Weill
err分享
err收藏
A survey of numerical linear algebra methods utilizing mixed-precision arithmetic
err2021-03-19
err80
PREAI
errAbdelfattah, Ahmad; Anzt, Hartwig; Boman, Erik G.; Carson, Erin; Cojean, Terry; Dongarra, Jack; Fox, Alyson; Gates, Mark; Higham, Nicholas J.; Li, Xiaoye S.; Loe, Jennifer; Luszczek, Piotr; Pranesh, Srikara; Rajamanickam, Siva; Ribizel, Tobias; Smith, Barry F.; Swirydowicz, Kasia; Thomas, Stephen; Tomov, Stanimire; Tsai, Yaohung M.; Yang, Ulrike Meier
err分享
err收藏
Preclinical Animal Models of Autistic Spectrum Disorders (ASD)
err2008-01-01
err0
PREAI
errJennifer A. Bartz; Larry J. Young; Eric Hollander; Joseph D. Buxbaum; Robert H. Ring
err分享
err收藏
Variability of electrophrenic diaphragm twitch stimulation over time in normal subjects
err1999-10-01
err0
PREAI
errGerard J Criner; John M Travaline; Greg A Holt; Christopher G Bosse; Steven G Kelsen
err分享
err收藏
High-Performance Tensor Learning Primitives Using GPU Tensor Cores使用GPU张量核的高性能张量学习原语
err2023-06-01
err3
PREAI
errLiu, Xiao-Yang; Zhang, Zeliang; Wang, Zhiyuan; Lu, Han; Wang, Xiaodong; Walid, Anwar
err分享
err收藏
Fine genetic mapping of the white immature fruit color gene w to a 33.0-kb region in cucumber (Cucumis sativus L.)
err2015-08-04
err0
PREAI
errHanqiang Liu; Huanwen Meng; Yupeng Pan; Xinjing Liang; Jianqing Jiao; Yuhong Li; Shuxia Chen; Zhihui Cheng
err分享
err收藏
err分享
err收藏
Do Real-Time Strategy Video Gamers Have Better Attentional Control?
err2022-01-10
err0
PREAI
errMengxin He; Lin-Xuan Xu; Chiang-shan R. Li; Zihan Liu; Jiaqi Hu; Xiangyi Guo; Hongyun Liu; Jin-Tao Zhang
err分享
err收藏
学者 查看更多内容