* Use black for formatting
* limit line width to 80 characters.
* Use a backslash instead of directly concatenating
* file structure adjustment.
* file structure adjustment(2)
* codes for citation2
* format slight adjustment
* adjust format in models.py
* now it runs normally for all datasets.
* add comments; adjust code order.
* adjust indenting.
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
* Use black for formatting
* limit line width to 80 characters.
* Use a backslash instead of directly concatenating
* file structure adjustment.
* file structure adjustment(2)
* codes for citation2
* format slight adjustment
* adjust format in models.py
* now it runs normally for all datasets.
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
* Add device support in hetero_rgcn
Signed-off-by: Xue, Chendi <chendi.xue@intel.com>
* use num_workers instead of hard code and enable cpu_affinity for pytorch > 1.12
Signed-off-by: Xue, Chendi <chendi.xue@intel.com>
* Remove hard code num_workers and use dgl version to check if cpu_affinity is supported
Signed-off-by: Xue, Chendi <chendi.xue@intel.com>
* Remove specified dl_cores and computer_cores
Add error print if num_workers are miss set
Signed-off-by: Xue, Chendi <chendi.xue@intel.com>
* expected num_workers should be less than num_physical_cores
Signed-off-by: Chendi Xue <chendi.xue@intel.com>
* Update examples/pytorch/ogb/ogbn-mag/hetero_rgcn.py
Co-authored-by: Chang Liu <chang.liu@utexas.edu>
* Remove dgl version and num_workers print
Signed-off-by: Xue, Chendi <chendi.xue@intel.com>
* add comment to explain is_support_affinity
* Fix typo
Signed-off-by: Xue, Chendi <chendi.xue@intel.com>
Signed-off-by: Xue, Chendi <chendi.xue@intel.com>
Signed-off-by: Chendi Xue <chendi.xue@intel.com>
Co-authored-by: Chang Liu <chang.liu@utexas.edu>
* Use black for formatting
* limit line width to 80 characters.
* Use a backslash instead of directly concatenating
* file structure adjustment.
* file structure adjustment(2)
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
* NGNN for ogbl
* modify doc organization.
* merge similar parts
* 1st approving review.
* minor changes
* Remove the "Usage" section.
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
* rgcn-ogbn-mag
* Add link in README.md
* correct code-format,add the reset_parameters function to the HeteroEmbedding module
* add the annotation in hetero.py
* add a unit test
* modify format
* Update
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-50-143.us-west-2.compute.internal>
* Adding initial files of example
* Removing old timing code
* Improving doc strings and fixing some minor bugs
* Merging from upstream and addressing PR comments
Co-authored-by: zakjost <jostza@amazon.com>
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
* [Example] GCN on ogbn-arxiv dataset
* Add README.md
* Update GCN implementation on ogbn-arxiv
* Update GCN on ogbn-arxiv
* Fix typo
* Use evaluator to get results
* Fix duplicated
* Fix duplicated
* Update GCN on ogbn-arxiv
* Add GAT for ogbn-arxiv
* Update README.md
* Update GAT on ogbn-arxiv
* Update README.md
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
Co-authored-by: Zihao Ye <expye@outlook.com>
* [Example] GCN on ogbn-arxiv dataset
* Add README.md
* Update GCN implementation on ogbn-arxiv
* Update GCN on ogbn-arxiv
* Fix typo
* Use evaluator to get results
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
Co-authored-by: Zihao Ye <expye@outlook.com>
* mx tf relconv
* use method instead of private attr
* src and dst have different fc for gat
* update edgeconv
* change sage and sgconv
* no degree check on gin
* add remainding API doc
* fix pylint
* infer fc_src and fc_dst, only one tensor for block
* fix pytest
* clean commit
* oops forgot the most important files
* use einsum
* copy feature from frontier to block
* Revert "copy feature from frontier to block"
This reverts commit 5224ec963eb6a3ef1b6ab74d8ecbd44e4e42f285.
* temp fix
* unit test
* fix
* revert jtnn
* lint
* fix win64
* docstring fixes and doc indexing
* revert einsum in sparse bidecoder
* fix some examples
* lint
* fix due to some tediousness in remove_edges
* addresses comments
* fix
* more jtnn fixes
* fix
Chang Liu