Search Results for author: Ehsan K. Ardestani

Found 4 papers, 0 papers with code

MTrainS: Improving DLRM training efficiency using heterogeneous memories

no code implementations19 Apr 2023 Hiwot Tadese Kassa, Paul Johnson, Jason Akers, Mrinmoy Ghosh, Andrew Tulloch, Dheevatsa Mudigere, Jongsoo Park, Xing Liu, Ronald Dreslinski, Ehsan K. Ardestani

In Deep Learning Recommendation Models (DLRM), sparse features capturing categorical inputs through embedding tables are the major contributors to model size and require high memory bandwidth.

Building a Performance Model for Deep Learning Recommendation Model Training on GPUs

no code implementations19 Jan 2022 Zhongyi Lin, Louis Feng, Ehsan K. Ardestani, Jaewon Lee, John Lundell, Changkyu Kim, Arun Kejariwal, John D. Owens

We show that our general performance model not only achieves low prediction error on DLRM, which has highly customized configurations and is dominated by multiple factors but also yields comparable accuracy on other compute-bound ML models targeted by most previous methods.

Cannot find the paper you are looking for? You can Submit a new open access paper.