Characterizing Matrix Multiplication Units across General Parallel Patterns in Scientific Computing

View at publisher
Authors: Yuechen Lu, Hongwei Zeng, Marc Casas, Weifeng Liu
Cited by: 1
References: 77

Abstract

Matrix multiplication units (MMUs) in modern parallel processors enable efficient execution of tiled matrix multiplications at varying precisions. While their effectiveness in AI workloads has been well demonstrated, their utility in scientific computing lacks systematic analysis. In this work, we characterize MMUs across a broad range of scientific computing patterns by evaluating performance, power consumption, numerical precision, and memory access behavior. To support this analysis, we develop Cubie, a comprehensive benchmark suite comprising ten MMU-optimized kernels of key parallel patterns. We also categorize MMU utilization patterns into four quadrants and identify the MMU limitations that arise in scientific computing. Through detailed comparisons with vector units, we provide nine key observations on the behavior and implications of MMUs in general scientific workloads, offering valuable insights for architecture, algorithm, and application researchers.

Fields of Study & Subjects

Parallel Computing and Optimization Techniques Interconnection Networks and Systems Advanced Data Storage Technologies

Publication details

Journal / source
Proceedings of the 31st ACM SIGPLAN Annual Symposium on Principles and Practice of Parallel Programming
Publisher
ACM
Pages
687–701
Publication date
28 January 2026