Search Results for author:
Found 5 papers, 4 papers with code
LightSeq2: Accelerated Training for Transformer-based Models on GPUs
In this paper, we present LightSeq2, a system to accelerate training for a general family of Transformer models on GPUs.
GE-SpMM: General-purpose Sparse Matrix-Matrix Multiplication on GPUs for Graph Neural Networks
GE-SpMM performs SpMM-like operation on sparse matrices represented in the most common Compressed Sparse Row (CSR) format, so it can be embedded in GNN frameworks with no preprocessing overheads and support general GNN algorithms.
Distributed, Parallel, and Cluster Computing
TC-GNN: Bridging Sparse GNN Computation and Dense Tensor Cores on GPUs
Recently, graph neural networks (GNNs), as the backbone of graph-based machine learning, demonstrate great success in various domains (e. g., e-commerce).
Machine Learning for Electronic Design Automation: A Survey
With the down-scaling of CMOS technology, the design complexity of very large-scale integrated (VLSI) is increasing.
Understanding GNN Computational Graph: A Coordinated Computation, IO, and Memory Perspective
The same data are propagated through the graph structure to perform the same neural operation multiple times in GNNs, leading to redundant computation which accounts for 92. 4% of total operators.
