Saved in:
Bibliographic Details
Main Authors: Wang, Hansheng, Shi, Lu, duan, Zhekai, Wu, Panruo, Guo, Liwei, Zhang, Shaoshuai
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2410.02170
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916421718507520
author Wang, Hansheng
Shi, Lu
duan, Zhekai
Wu, Panruo
Guo, Liwei
Zhang, Shaoshuai
author_facet Wang, Hansheng
Shi, Lu
duan, Zhekai
Wu, Panruo
Guo, Liwei
Zhang, Shaoshuai
contents Benefiting from the advancement of hardware accelerators such as GPUs, deep neural networks and scientific computing applications can achieve superior performance. Recently, the computing capacity of emerging hardware accelerators has increased rapidly, while memory bandwidth has not kept pace with this growth. This disparity exacerbates the gap between computing and memory, leading to inefficiencies on conventional algorithms, as they're likely to be converted from compute-bound to memory-bound. Symmetric eigenvalue decomposition (EVD), a critical operation in various research domains including scientific computing, deep learning training, and inference algorithms, exhibits suboptimal performance due to achieving less than 3\% hardware computing utilization on the H100 GPU. In this paper, we analyze the features of emerging hardware accelerators to identify the bottlenecks inherent in conventional EVD algorithms. To improve EVD performance, we propose several algorithmic optimizations aimed at solving the memory-bound problem and providing a better utilization of the rich computing capacity and parallelism on the emerging hardware accelerators. Experimentally, our proposed method demonstrates significant speedups on tridiagonalization, which is the main workload that takes over 90\% elapsed time of EVD, compared to the SOTA cuSOLVER tridiagonalization, achieving up to 10.1x, 7.5x, and 2.3x improvements on H100, A100, and RTX 4090 GPUs, respectively. And the end-to-end the performance of EVD solver is also up to 4.1x faster than cuSOVLER.
format Preprint
id arxiv_https___arxiv_org_abs_2410_02170
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Extracting the Potential of Emerging Hardware Accelerators for Symmetric Eigenvalue Decomposition
Wang, Hansheng
Shi, Lu
duan, Zhekai
Wu, Panruo
Guo, Liwei
Zhang, Shaoshuai
Distributed, Parallel, and Cluster Computing
Benefiting from the advancement of hardware accelerators such as GPUs, deep neural networks and scientific computing applications can achieve superior performance. Recently, the computing capacity of emerging hardware accelerators has increased rapidly, while memory bandwidth has not kept pace with this growth. This disparity exacerbates the gap between computing and memory, leading to inefficiencies on conventional algorithms, as they're likely to be converted from compute-bound to memory-bound. Symmetric eigenvalue decomposition (EVD), a critical operation in various research domains including scientific computing, deep learning training, and inference algorithms, exhibits suboptimal performance due to achieving less than 3\% hardware computing utilization on the H100 GPU. In this paper, we analyze the features of emerging hardware accelerators to identify the bottlenecks inherent in conventional EVD algorithms. To improve EVD performance, we propose several algorithmic optimizations aimed at solving the memory-bound problem and providing a better utilization of the rich computing capacity and parallelism on the emerging hardware accelerators. Experimentally, our proposed method demonstrates significant speedups on tridiagonalization, which is the main workload that takes over 90\% elapsed time of EVD, compared to the SOTA cuSOLVER tridiagonalization, achieving up to 10.1x, 7.5x, and 2.3x improvements on H100, A100, and RTX 4090 GPUs, respectively. And the end-to-end the performance of EVD solver is also up to 4.1x faster than cuSOVLER.
title Extracting the Potential of Emerging Hardware Accelerators for Symmetric Eigenvalue Decomposition
topic Distributed, Parallel, and Cluster Computing
url https://arxiv.org/abs/2410.02170