oneflow icon indicating copy to clipboard operation
oneflow copied to clipboard

Dev gcn spmm [don't merge]

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

This PR records the progress relating to GCN and SpMM. A prior PR is #9092.

Two components are added:

  1. spmm op & kernels
  2. gcn model

The gcn model is implemented based on SpMM. Currently, it allows (1) mini-batch training based graph sampling and (2) full-batch training OneEmbedding.

The (unfinished) goal is to achieve mini-batch training with OneEmbedding.

The detailed explanantions can be found in /gcn folder.

yuang-chen avatar Nov 17 '22 13:11 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 Nov 17 '22 13:11 CLAassistant