中文

网内累加:扩展NoC在DNN加速中的作用

硬件体系结构 2022-09-22 v1 分布式、并行与集群计算

摘要

片上网络(Network-on-Chip, NoC)对DNN加速器的性能起着重要作用。NoC的可扩展性与模块化设计特性,通过提供运行不同工作负载的灵活性,有助于提升DNN执行性能。DNN工作负载中的数据搬移对DNN加速器而言仍是挑战性任务,因而需要一种新方法。本文提出网内累加(In-Network Accumulation, INA)方法,以在权重固定(Weight Stationary, WS)数据流模型的多核空间DNN加速器上进一步加速DNN工作负载执行。INA方法扩展路由器功能以支持部分和累加,避免了将传入部分和注入并弹出至本地处理单元的开销。在AlexNet、ResNet-50与VGG-16工作负载上的仿真结果表明,所提INA方法在WS数据流模型上实现延迟1.22倍提升与功耗2.16倍改善。

关键词

引用

@article{arxiv.2209.10056,
  title  = {In-Network Accumulation: Extending the Role of NoC for DNN Acceleration},
  author = {Binayak Tiwari and Mei Yang and Xiaohang Wang and Yingtao Jiang},
  journal= {arXiv preprint arXiv:2209.10056},
  year   = {2022}
}

备注

Presented at 35th IEEE International System-on-Chip Conference (SOCC)