Understanding Reuse, Performance, and Hardware Cost of DNN Dataflow: A Data-Centric Approach

By Hyoukjun Kwon, Prasanth Chatarasi, Michael Pellauer, Angshuman Parashar, Vivek Sarkar, Tushar Krishna. Micro, 2018

349 citations recorded by the source corpus.

The data partitioning and scheduling strategies used by DNN accelerators to leverage reuse and perform staging are known as dataflow, which directly impacts the performance and energy efficiency of DNN accelerators. An accelerator micro architecture dictates the dataflow(s) that can be employed to execute layers in a DNN. Selecting a dataflow for a layer can have a large impact on utilization and energy efficiency, but there is a lack of understanding on the choices and consequences of dataflow, and of tools and methodologies to help architects explore the co-optimization design space. In this work, we first introduce a set of data-centric directives to concisely specify the DNN dataflow space in a compiler-friendly form. We then show how these directives can be analyzed to infer various forms of reuse and to exploit them using hardware capabilities. We codify this analysis into an analytical cost model, MAESTRO (Modeling Accelerator Efficiency via Patio-Temporal Reuse and Occupancy), that estimates various cost-benefit tradeoffs of a dataflow including execution time and energy efficiency for a DNN model and hardware configuration. We demonstrate the use of MAESTRO to drive a hardware design space exploration experiment, which searches across 480M designs to identify 2.5M valid designs at an average rate of 0.17M designs per second, including Pareto-optimal throughput- and energy-optimized design points.

Subfields: Computer Networks and Communications, Electrical and Electronic Engineering, Hardware and Architecture

View Paper: https://www.semanticscholar.org/paper/1f0bbcbcea15b60b39012e9aedf4dac42dff9411

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Computer Networks and Communications

  1. Shuai Che
  2. Tushar Krishna
  3. Nam Sung Kim
  4. Hadi Esmaeilzadeh
  5. Steven K. Reinhardt
  6. Michael Nicolaidis

In Electrical and Electronic Engineering

  1. Tushar Krishna
  2. Hyoukjun Kwon
  3. Nam Sung Kim
  4. Hadi Esmaeilzadeh
  5. Steven K. Reinhardt
  6. Joon Kyung Kim
  7. Hardik Sharma
  8. A. Samajdar

In Hardware and Architecture

  1. Nam Sung Kim
  2. Shuai Che
  3. Tushar Krishna
  4. Hadi Esmaeilzadeh
  5. Steven K. Reinhardt
  6. Cliff Young
  7. Sparsh Mittal
  8. Michael Nicolaidis

Citations in the corpus