English

Adding MFMA Support to gem5

Hardware Architecture 2025-02-04 v2

Abstract

In this work we have enhanced gem5's GPU model support to add Matrix Core Engines (MCEs). Specifically, on the AMD MI200 and MI300 GPUs that gem5 supports, these MCEs perform Matrix Fused Multiply Add (MFMA) instructions for a variety of precisions. By adding this support, our changes enable running state-of-the-art ML workloads in gem5, as well as examining how MCE optimizations impact the behavior of future systems.

Cite

@article{arxiv.2501.18113,
  title  = {Adding MFMA Support to gem5},
  author = {Marco Kurzynski and Matthew D. Sinclair},
  journal= {arXiv preprint arXiv:2501.18113},
  year   = {2025}
}

Comments

7 pages

R2 v1 2026-06-28T21:24:58.801Z