Accelerating SpMM Kernel with Cache-First Edge Sampling for Graph Neural Networks

Chien-Yu Lin, Liang Luo, Luis Ceze

Graph neural networks (GNNs), an emerging deep learning model class, can extract meaningful representations from highly expressive graph-structured data and are therefore gaining p…