arrow
Return

Simba: Scaling Deep-Learning Inference with Chiplet-Based Architecture

delete2021-05-24
delete10
delete
OA
AI
Y
Yakun Sophia Shao *
J
Jason Cemons
R
Rangharajan Venkatesan
B
Brian Zimmer
M
Matthew Fojtik
N
Nan Jiang
B
Ben Keller
A
Alicia Klinefelter
N
Nathaniel Pinckney
P
Priyanka Raina
S
Stephen G. Tell
张燕青 cover
张燕青 (Yanqing Zhang)
W
William J. Dally
J
Joel Emer
C
C. Thomas Gray
B
Brucek Khailany
S
Stephen W. Keckler
DOI:10.1145/3460227delete
deleteOriginal
deleteShare
deleteSave
View PDF
Abstract

Abstract

En 中文
Package-level integration using multi-chip-modules (MCMs) is a promising approach for building large-scale systems. Compared to a large monolithic die, an MCM combines many smaller chiplets into a larger system, substantially reducing fabrication and design costs. Current MCMs typically only contain a handful of coarse-grained large chiplets due to the high area, performance, and energy overheads associated with inter-chiplet communication. This work investigates and quantifies the costs and benefits of using MCMs with fine-grained chiplets for deep learning inference, an application domain with large compute and on-chip storage requirements. To evaluate the approach, we architected, implemented, fabricated, and tested Simba, a 36-chiplet prototype MCM system for deep-learning inference. Each chiplet achieves 4 TOPS peak performance, and the 36-chiplet MCM package achieves up to 128 TOPS and up to 6.1 TOPS/W. The MCM is configurable to support a flexible mapping of DNN layers to the distributed compute and storage units. To mitigate inter-chiplet communication overheads, we introduce three tiling optimizations that improve data locality. These optimizations achieve up to 16% speedup compared to the baseline layer mapping. Our evaluation shows that Simba can process 1988 images/s running ResNet-50 with a batch size of one, delivering an inference latency of 0.50 ms.
AI Summary

AI Summary

Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.

Journal

Communications of the ACM cover
Communications of the ACM
IF:
12.2
Papers:
1.2W
Citations:
3.7W

Organization

U
University of California Berkeley
Scholars:
3.5W
Papers: 2.8W
Citations: 11.3W
S
Stanford University
Scholars:
9.6W
Papers: 8.2W
Citations: 17.0W
N
nvidia corporation
Scholars:
767
Papers: 439
Citations: 1
University of California System cover
University of California System
Scholars:
37.5W
Papers: 33.7W
Citations: 6.6K
researcher View more organizations