TC-GNN_ATC23 icon indicating copy to clipboard operation
TC-GNN_ATC23 copied to clipboard

Published 20 hours ago • verified publisher icon YukeWang96

spmm kernel

Open aGoodCat opened this issue 1 year ago • 3 comments

Hi Yuke, What is the interface to be used to just perform SpMM? Thanks

aGoodCat avatar May 28 '23 03:05 aGoodCat

Please follow the instructions in this subsection. https://github.com/YukeWang96/TC-GNN_ATC23#8-running-tc-gnn-in-single-kernel-comparison

YukeWang96 avatar May 28 '23 03:05 YukeWang96

Hello Yuke, Does the "TC-GNN in single-kernel comparison" only run one SpMM? How can I change the feature size of SpMM? Let's say we have a sparse matrix which is MK, and we want to do SpMM on a dense KN matrix. How can I set the N? I see in your python script, I can pass "dim" and "hidden" as arguments. Which one defines the output feature size(N)? If it just run one SpMM, why it needs to parameters? Thank you so much

black-cat-sheriff avatar Jul 02 '23 05:07 black-cat-sheriff

Hi,

Single-kernel profiling uses the dataset.dim as the output feature size (N), where input and output has the same dimension. https://github.com/YukeWang96/TC-GNN_ATC23/blob/f50ffc491fb07bc78b7af8c0ffdec2b0bd7ec1a2/main_tcgnn.py#L35

https://github.com/YukeWang96/TC-GNN_ATC23/blob/f50ffc491fb07bc78b7af8c0ffdec2b0bd7ec1a2/gnn_conv.py#L179

YukeWang96 avatar Jul 03 '23 04:07 YukeWang96