GE-SpMM: General-purpose Sparse Matrix-Matrix Multiplication on GPUs for Graph Neural Networks.

Guyue Huang Guohao Dai Yu Wang Huazhong Yang

Published in: CoRR (2020)

Keyphrases

general purpose
sparse matrix
matrix multiplication
neural network
pattern recognition
message passing
random projections
graphics processing units
distributed memory
active learning
matrix factorization
rows and columns
high quality
data points
graph partitioning
linear equations