A copy of this work was available on the public web and has been preserved in the Wayback Machine. The capture dates from 2020; you can also visit the original URL.
The file type is application/pdf.
Filters
GE-SpMM: General-purpose Sparse Matrix-Matrix Multiplication on GPUs for Graph Neural Networks
[article]
2020
arXiv
pre-print
Preserved Fulltext
Graph Neural Networks (GNNs) have achieved significant improvements in various domains. ...
Sparse Matrix-Matrix multiplication (SpMM) is a fundamental operator in GNNs, which performs a multiplication between a sparse matrix and a dense matrix. ...
CONCLUSION In this paper, we propose an efficient CSR-based SpMM design, GE-SpMM, for Graph Neural Network applications on GPUs. ...
arXiv:2007.03179v1
fatcat:epshcpa7fbbdbcjchdkrocej3m
Computing Graph Neural Networks: A Survey from Algorithms to Accelerators
[article]
2021
arXiv
pre-print
Preserved Fulltext
A copy of this work was available on the public web and has been preserved in the Wayback Machine. The capture dates from 2021; you can also visit the original URL.
The file type is application/pdf.
Other Versions
Graph Neural Networks (GNNs) have exploded onto the machine learning scene in recent years owing to their capability to model and learn from graph-structured data. ...
Such an ability has strong implications in a wide variety of fields whose data is inherently relational, for which conventional neural networks do not perform well. ...
ACKNOWLEDGMENTS The authors would like to thank the anonymous reviewers and the editorial team for their constructive criticism, which has helped improve the quality of the paper. ...
arXiv:2010.00130v3
fatcat:u5bcmjodcfdh7pew4nssjemdba
FusedMM: A Unified SDDMM-SpMM Kernel for Graph Embedding and Graph Neural Networks
[article]
2021
arXiv
pre-print
Preserved Fulltext
A copy of this work was available on the public web and has been preserved in the Wayback Machine. The capture dates from 2020; you can also visit the original URL.
The file type is application/pdf.
Note that this fulltext copy is not of the "primary" version of this work. The version it corresponds to is:
2020 pre-print arXiv:2011.06391v1 fatcat:6s5dnzdt7zbdtdbho3kmmcnu64
We develop a fused matrix multiplication kernel that unifies sampled dense-dense matrix multiplication and sparse-dense matrix multiplication under a single operation called FusedMM. ...
FusedMM can tune its performance using a code generator and perform equally well on Intel, AMD and ARM processors. ...
[17] developed a general-purpose SpMM algorithm for GPUs. ...
arXiv:2011.06391v2
fatcat:3qh7paknwfbn7mb7vg6z3bltia