MT4G: A Tool for Reliable Auto-Discovery of NVIDIA and AMD GPU Compute and Memory Topologies
Stepan Vanecek, Manuel Walter Mussbacher, Dominik Groessler, Urvij Saroliya, Martin Schulz
TL;DR
MT4G tackles the lack of standardized GPU topology information by delivering an open-source, vendor-agnostic tool that auto-discovers compute and memory topologies for NVIDIA and AMD GPUs. It combines API data with a large microbenchmark suite and Kolmogorov-Smirnov–based change-point detection to reliably estimate topology attributes such as cache sizes, latencies, bandwidths, and fetch granularity. The approach is validated across ten GPUs and demonstrated in three workflows: GPU performance modeling, GPUscout bottleneck analysis, and dynamic resource partitioning with sys-sage, highlighting MT4G's practical impact on performance optimization and resource management in HPC/AI workloads. By providing automated, portable topology reports, MT4G enables more accurate modeling, tuning, and runtime configuration decisions across vendor platforms.
Abstract
Understanding GPU topology is essential for performance-related tasks in HPC or AI. Yet, unlike for CPUs with tools like hwloc, GPU information is hard to come by, incomplete, and vendor-specific. In this work, we address this gap and present MT4G, an open-source and vendor-agnostic tool that automatically discovers GPU compute and memory topologies and configurations, including cache sizes, bandwidths, and physical layouts. MT4G combines existing APIs with a suite of over 50 microbenchmarks, applying statistical methods, such as the Kolmogorov-Smirnov test, to automatically and reliably identify otherwise programmatically unavailable topological attributes. We showcase MT4G's universality on ten different GPUs and demonstrate its impact through integration into three workflows: GPU performance modeling, GPUscout bottleneck analysis, and dynamic resource partitioning. These scenarios highlight MT4G's role in understanding system performance and characteristics across NVIDIA and AMD GPUs, providing an automated, portable solution for modern HPC and AI systems.
