Travel Time Based Task Mapping for NoC-Based DNN Accelerator

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Chen, Yizhi, Zhu, Wenyao, Lu, Zhonghai
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915471784148992
author Chen, Yizhi
Zhu, Wenyao
Lu, Zhonghai
author_facet Chen, Yizhi
Zhu, Wenyao
Lu, Zhonghai
contents Network-on-Chip (NoC) based architectures are recently proposed to accelerate deep neural networks in specialized hardware. Given that the hardware configuration is fixed post-manufacture, proper task mapping attracts researchers' interest. We propose a travel time-based task mapping method that allocates uneven counts of tasks across different Processing Elements (PEs). This approach utilizes the travel time recorded in the sampling window and implicitly makes use of static NoC architecture information and dynamic NoC congestion status. Furthermore, we examine the effectiveness of our method under various configurations, including different mapping iterations, flit sizes, and NoC architecture. Our method achieves up to 12.1% improvement compared with even mapping and static distance mapping for one layer. For a complete NN example, our method achieves 10.37% and 13.75% overall improvements to row-major mapping and distance-based mapping, respectively. While ideal travel time-based mapping (post-run) achieves 10.37% overall improvements to row-major mapping, we adopt a sampling window to efficiently map tasks during the running, achieving 8.17% (sampling window 10) improvement.
format Preprint
id arxiv_https___arxiv_org_abs_2411_12710
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Travel Time Based Task Mapping for NoC-Based DNN Accelerator
Chen, Yizhi
Zhu, Wenyao
Lu, Zhonghai
Hardware Architecture
Network-on-Chip (NoC) based architectures are recently proposed to accelerate deep neural networks in specialized hardware. Given that the hardware configuration is fixed post-manufacture, proper task mapping attracts researchers' interest. We propose a travel time-based task mapping method that allocates uneven counts of tasks across different Processing Elements (PEs). This approach utilizes the travel time recorded in the sampling window and implicitly makes use of static NoC architecture information and dynamic NoC congestion status. Furthermore, we examine the effectiveness of our method under various configurations, including different mapping iterations, flit sizes, and NoC architecture. Our method achieves up to 12.1% improvement compared with even mapping and static distance mapping for one layer. For a complete NN example, our method achieves 10.37% and 13.75% overall improvements to row-major mapping and distance-based mapping, respectively. While ideal travel time-based mapping (post-run) achieves 10.37% overall improvements to row-major mapping, we adopt a sampling window to efficiently map tasks during the running, achieving 8.17% (sampling window 10) improvement.
title Travel Time Based Task Mapping for NoC-Based DNN Accelerator
topic Hardware Architecture
url https://arxiv.org/abs/2411.12710