Accelerating Multi-GPU Embedding Retrieval with PGAS-Style Communication for Deep Learning Recommendation Systems

In this paper, we propose using Partitioned Global Address Space (PGAS) GPU one-sided asynchronous small messages to replace the widely used collective communication calls for sparse input multi-GPU embedding retrieval in deep learning recommendation systems. This GPU PGAS communication approach ach...

Full description

Bibliographic Details

Published in:	SC24-W: Workshops of the International Conference for High Performance Computing, Networking, Storage and Analysis pp. 1262 - 1273
Main Authors:	Chen, Yuxin, Buluc, Aydin, Yelick, Katherine, Owens, John D.
Format:	Conference Proceeding
Language:	English
Published:	IEEE 17.11.2024
Subjects:	Analytical models Collective Calls Communication Computational modeling Conferences Deep learning Deep Learning Recommendation Model Electronics packaging Graphics processing units High performance computing PGAS Recommender systems
Online Access:	Get full text
Tags:	Add Tag No Tags, Be the first to tag this record!