Dissecting the NVIDIA Hopper Architecture through Microbenchmarking and Multiple Level Analysis

AI Systems and Hardware TPU/NPU EDIO
现代GPU,凭借其专门的硬件如张量核心,对于高性能的人工智能和深度学习应用至关重要。本研究对NVIDIA Hopper GPU架构进行了全面的多层次微基准测试分析,深入探讨了其性能特点和新特性。我们测试了Hopper内存子系统的延迟和吞吐量,将其L2分区分级缓存行为和全局内存访问模式与最近的Ampere和Ada Lovelace两代GPU进行了比较。分析结果显示Hopper在性能差异和架构改进方面有显著提升。本工作的核心贡献是对Hopper第四代张量核心的详细评估,包括其FP8精度支持和新型异步wgmma指令,并评估这些因素对矩阵乘积累加操作的影响。我们进一步研究了其他关键Hopper创新对性能的影响:用于加速动态规划算法的DPX指令、用于SM间通信的分布式共享内存(DSM),以及用于异步数据移动的张量内存加速器(TMA)。这一多层次方法涵盖了指令级别的微基准测试、Transformer Engine库级别的分析,以及在大型语言模型中对张量核心性能的应用级别基准测试。我们的研究结果为软件开发者提供了宝贵而深入的见解,有助于优化Hopper架构的性能并开发准确的性能模型,最终加深对其在加速人工智能和其他计算密集型工作负载方面的潜力的理解。
Modern GPUs, with their specialized hardware like tensor cores, are essential for demanding AI and deep learning applications. This study presents a comprehensive, multi-level microbenchmarking analysis of the NVIDIA Hopper GPU architecture, delving into its performance characteristics and novel features. We benchmark Hopper's memory subsystem latency and throughput, comparing its L2 partitioned cache behavior and global memory access patterns against recent GPU generations, Ampere and Ada Lovelace. Our analysis reveals significant performance differences and architectural improvements in Hopper. A core contribution of this work is a detailed evaluation of Hopper's fourth-generation tensor cores, including their FP8 precision support and the novel asynchronous wgmma instructions, assessing their impact on matrix multiply-accumulate operations. We further investigate the performance implications of other key Hopper innovations: DPX instructions for accelerating dynamic programming algorithms, distributed shared memory (DSM) for inter-SM communication, and the Tensor Memory Accelerator (TMA) for asynchronous data movement. This multi-level approach encompasses instruction-level microbenchmarks, library-level analysis of the Transformer Engine, and application-level benchmarks of tensor core performance within large language models. Our findings provide valuable, in-depth insights for software developers seeking to optimize performance and develop accurate performance models for the Hopper architecture, ultimately contributing to a deeper understanding of its potential for accelerating AI and other computationally intensive workloads.
许愿