Brief Announcement: Optimality Conditions for Parallel Communication-Avoiding Matrix Multiplication with Overlapped Communication
When considering general matrix multiply (GEMM) algorithms for distributed-memory systems, the dominant paradigm is to minimize communication volume. However, to minimize time, one must consider how communication volume interacts with the system characteristics --- namely, communication bandwidth, memory capacity, and the co-scheduling of computation and communication. In this work, we demonstrate that the family of 3D GEMM algorithms --- although reducing overall communication by leveraging extra memory --- fundamentally concentrates communication to the phase of memory filling. This upfront cost hinders the ability to overlap communication with computation, and, consequently, calls for a revised view of the GEMM optimality regions. Our main contribution is derivation of optimality conditions for parallel matrix multiplication that jointly accounts for communication volume and overlap effects, pushing the 3D GEMM optimality region to start with the systems more than 11× larger than previously believed.
Authors
- Richard Vuduc (ORCID: https://orcid.org/0000-0003-2178-138X)
- Srinivas Eswar (ORCID: https://orcid.org/0000-0002-3418-7796)
- Mikhail Isaev (ORCID: https://orcid.org/0000-0002-2599-9741)
Institutions
- Argonne National Laboratory (US)
- Nvidia (United Kingdom) (GB)
- Georgia Institute of Technology (US)
- Nvidia (United States) (US)
Publication Details
- Published
- 2025-07-16
- DOI
- https://doi.org/10.1145/3694906.3743357
- Primary Topic
- Interconnection Networks and Systems
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- National Science Foundation
- U.S. Department of Energy
- Office of Science
- Argonne National Laboratory