Abstract: With the advancement of Artificial Intelligence (AI), the reliability of AI accelerators has become increasingly critical. Moreover, sparse matrix multiplication has become a fundamental ...

Abstract: Sparse Matrix-Matrix Multiplication (SpMM) is a widely used algorithm in Machine Learning, particularly in the increasingly popular Graph Neural Networks (GNNs). SpMM is an essential ...

Currently many operations in wp.sparse modify the end matrix topology, using CUB-backed reductions that require temporary storage allocations under the hood. As a result, then cannot be captured in ...