TC-GNN

Table of Contents

paper link: https://arxiv.org/pdf/2112.02052.pdf

TL;DR: This paper propose to co-densing (SpArch also propose co-densing) the sparse matrix \(A\) in SpMM \(C=A\times B\) to reduce the fragmentation when using Tensor Cores for SpMM computations on GPU.

2022-08-17_14-26-08_screenshot.png

My Digest

The co-densing could be described as a special format rewrite rules in SparseTIR, worth trying it and add an example in the documentation.

Author: expye(Zihao Ye)

Email: expye@outlook.com

Date: 2022-08-17 Wed 00:00

Last modified: 2022-09-23 Fri 00:04

Licensed under CC BY-NC 4.0