oneflow icon indicating copy to clipboard operation
oneflow copied to clipboard

add spmm_coo op & kernel (as a part of GCN)

Open yuang-chen opened this issue 2 years ago • 1 comments

SpMM COO

NOT ready for merging!

Add the basic spmm_coo to perform SpMM operations needed by GCN.

Only the CUDA kernel is offered.

Code pieces are borrowed from prior oneflow-GCN implementation.

GCN

GCN is implemented using the spmm_coo op, denoted as CooGCN. Compared with the PyGCN by Kipf, the performance of CooGCN running on Cora graph with 200 epochs is:

CooGCN PyGCN
Time 1.2032s 1.5967s
Loss 0.7060 0.7612
accuracy 0.8070 0.8150

yuang-chen avatar Sep 14 '22 10:09 yuang-chen

CLA assistant check
Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you sign our Contributor License Agreement before we can accept your contribution.
You have signed the CLA already but the status is still pending? Let us recheck it.

CLAassistant avatar Sep 14 '22 10:09 CLAassistant