mlsanityData to 14 Oct 2026

Exploring the Performance Improvement of Tensor Processing Engines through Transformation in the Bit-weight Dimension of MACs

arXiv 2503.06342 ↗First posted 8 Mar 2025cs.AR cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

General matrix-matrix multiplication (GEMM) is a cornerstone of AI computations, making tensor processing engines (TPEs) increasingly critical in GPUs and domain-specific architectures. Existing architectures primarily optimize dataflow or operand reuse strategies. However, considering the interaction between matrix multiplication and multiply-accumulators (MACs) offers greater optimization potential. This work introduces a novel hardware perspective on matrix multiplication, focusing on the bit-weight dimension of MACs. We propose a finer-grained TPE notation using matrix triple loops as an example, introducing new methods for designing and optimizing PE microarchitectures. Based on this notation and its transformations, we propose four optimization techniques that improve timing, area, and power consumption. Implementing our design in RTL using the SMIC-28nm process, we evaluate its effectiveness across four classic TPE architectures: systolic array, 3D-Cube, multiplier-adder tree, and 2D-Matrix. Our techniques achieve area efficiency improvements of 1.27x, 1.28x, 1.56x, and 1.44x, and energy efficiency gains of 1.04x, 1.56x, 1.49x, and 1.20x, respectively. Applied to a bit-slice architecture, our approach achieves a 12.10x improvement in energy efficiency and 2.85x in area efficiency compared to Laconic. Our Verilog HDL code, along with timing, area, and power reports, is available at https://github.com/wqzustc/High-Performance-Tensor-Processing-Engines

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 8 of its 51 references that are in the map. First posted on arXiv 8 Mar 2025.

Builds on

8 of 51 references.

References · 8

  1. 2017In-Datacenter Performance Analysis of a Tensor Processing Unitcited by 618
  2. 2023TPU v4: An Optically Reconfigurable Supercomputer for Machine Learning with Hardware Support for Embeddingscited by 202
  3. 2018NVIDIA Tensor Core Programmability, Performance & Precisioncited by 79
  4. 2019Dissecting the Graphcore IPU Architecture via Microbenchmarkingcited by 47
  5. 2018Interstellar: Using Halide's Scheduling Language to Analyze DNN Acceleratorscited by 34
  6. 2016Bit-pragmatic Deep Neural Network Computingcited by 32
  7. 2024Towards Cheaper Inference in Deep Networks with Lower Bit-Width Accumulatorscited by 4
  8. 2024EN-T: Optimizing Tensor Computing Engines Performance via Encoder-Based Methodologycited by 1

Built on by

Not cited yet.