提交

1593 次代码提交

作者 SHA1 备注 提交日期
Quan Gan e2e894c3f4 update version v0.6.0 2021-02-25 18:31:32 +00:00
Quan Gan e5aa149e80 update version 2021-02-25 18:31:10 +00:00
Quan Gan bbf4d8d4de update version 2021-02-25 18:31:10 +00:00
Ligeng Zhu 6568d0aaf6 [Example] Recommend NFS over copy files (#2689)
* Update README to use NFS, rather than copying files

* Update README.md

* Update README.md

* Update README.md

* Add ParMetis

* Update README.md

* fix directory.

* fix

* redo.

* fix.

* update.

* add more descriptions

Co-authored-by: Da Zheng <zhengda1936@gmail.com>
Co-authored-by: Quan Gan <coin2028@hotmail.com>
2021-02-25 20:20:52 +08:00
Quan (Andy) Gan 99751d4960 [Doc] Rename block to message flow graph (#2702)
* rename block to mfg

* revert

* rename
2021-02-25 18:26:19 +08:00
xnouhz 491d908b55 [Example] Add ARMA for Node Classification (#2694)
* [example] arma

* update

* update

* update

* update

* update

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2021-02-25 17:38:43 +08:00
Smile 583aa76621 [Example] Pytorch Seal example (#2638)
* add seal example

* 1. add paper infomation in examples/README
2. adjust codes
3. option test

* use latest `to_simple` to replace coalesce graph function

* remove outdated codes

* remove useless comment
2021-02-25 15:53:35 +08:00
Da Zheng 0526b885ee [Distributed] Fix the path for distributed partitioning. (#2700)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-9-132.us-west-1.compute.internal>
2021-02-24 20:41:41 -08:00
Ligeng Zhu 7ab3d993a4 Change the absolute path to relative one in part_config.json (#2690)
* Update partition.py

* fix lint.

Co-authored-by: Da Zheng <zhengda1936@gmail.com>
2021-02-24 16:12:19 -08:00
Quan (Andy) Gan 1686106390 change __repr__ (#2696)
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-02-25 00:21:00 +08:00
Jinjing Zhou 704ec68576 [Benchmark] Fix benchmark tests and add script to generate excel result (#2681)
* add script

* 21

* log

* fix regression tests

* add iter time

* fix

* Revert "fix"

This reverts commit 9b4587ad61dfeb82dcce6d1de188809f62947663.

* fix

* fix

* add

* add ogb instruction

* add cu11 dockerfile

* fix

* fix

* more iter apply_edge

* add iteration time

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-02-24 19:41:49 +08:00
Quan (Andy) Gan 71d53d9eef add thumbnails for tutorials (#2695)
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-02-23 20:33:35 +08:00
Quan (Andy) Gan da2f690a5a [Doc] Clarify link prediction tutorial (#2673)
* link prediction clarification

* fix
2021-02-23 19:41:54 +08:00
Tong He 7298173ac1 [Doc]Improve docstring for GINDataset (#2676)
* add docstring for gin

* update doc

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2021-02-23 15:14:10 +08:00
Zihao Ye 174dd18905 [hotfix] A bunch of fix for SetTransformer (#2658)
* upd

* lint

* upd

* fix

* fix

* upd

* upd

* fix

* warning

* upd

* upd
2021-02-23 14:19:51 +08:00
Da Zheng 48a1794f85 [Doc] Fix the user guide for distributed partitioning. (#2684)
* fix doc.

* explain the schema file.

* fix.
2021-02-22 18:59:15 -08:00
xiang song(charlie.song) 80c26877de [Bug Fix] Fix munmap bug in sparse optimizer (#2675)
* Fix munmap bug

* lint

* update

Co-authored-by: Ubuntu <ubuntu@ip-172-31-56-220.ec2.internal>
2021-02-22 11:33:44 +08:00
xiang song(charlie.song) 16e324daf3 [Bug Fix] Add warning for invalid input graph of edge UDF (#2682)
* Add warning for invalid graph of edge UDF

* lint

Co-authored-by: Ubuntu <ubuntu@ip-172-31-56-220.ec2.internal>
2021-02-22 10:12:40 +08:00
Zihao Ye 72a16b22e4 [bugfix] Fix the behavior of select_topk (#2688)
* fix

* lint

* lint
2021-02-21 18:52:35 +08:00
Zihao Ye e6bf54cdcb [Feature] Support aggregate multiple edge features in to_simple. (#2623)
* upd

* fix

* upd

* upd

* upd

* upd

* upd

* upd

* upd

* upd

* upd

* fix

* refactor

* upd test

* large feat_len or n in segment reduce

* lint
2021-02-21 10:05:42 +08:00
Minjie Wang 8a07ab7737 [Doc] Tutorials re-organization (#2683)
* reorg

* change titles

* rm some stale API doc; minor fix

* fix docs

* add warning

* rm new-tutorial run in ci

* lint
2021-02-20 17:42:45 +08:00
Da Zheng 0fc649523b [Doc] Distributed heterogeneous graph training (#2654)
* update DistGraph docstrings.

* add user guide.

* add doc string.

* fix.

* fix.

* fix.

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2021-02-19 10:48:05 -08:00
xiang song(charlie.song) 453d358d2c [Doc] Fix docs for sparse optimizer (#2680)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-56-220.ec2.internal>
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-02-19 14:07:35 +08:00
Quan (Andy) Gan 9e04a52abd [Doc] Reorganize tutorial (#2678) 2021-02-19 13:49:13 +08:00
Quan (Andy) Gan dda103d9f6 Fix bug in EdgeConv implementation (#2669)
* Update edgeconv.py

* update edgeconv to builtin

* add explanation on why ReLU is not there

* remove redundant code
2021-02-19 13:07:21 +08:00
Quan (Andy) Gan d93a975997 revert (#2672) 2021-02-19 12:05:42 +08:00
Da Zheng ff64bd0de8 add warning for a bug in README (#2665) 2021-02-16 00:16:40 -08:00
Zihao Ye 0cbaa1fbcb [Doc] Make the docstring of nn.glob more informative. (#2649)
* upd

* fix weight and sum

* upd sortpool
2021-02-12 12:29:22 +08:00
Quan (Andy) Gan 9e630101a1 fix and lots of tests (#2650) 2021-02-12 11:40:05 +08:00
Zihao Ye cf8a3fb305 [hotfix] Address the performance issue of topk. (#2628)
* upd

* fix

* upd

* upd

* udp

* upd

* upd

* upd

* upd

* upd

* upd

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2021-02-11 00:00:39 +08:00
Tianqi Zhang (张天启) e46983f662 [Bug Fix][Example] fix dataloader bug in gxn (#2631)
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-02-10 11:21:34 +08:00
Minjie Wang 22ccf4365a [Test] Tweak end2end benchmarks to be more reasonable (#2643)
* change timeout to reasonable ranges

* rgcn ns

* fix all ns speed tests

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2021-02-09 11:08:55 +08:00
Da Zheng e4ff484494 [Distributed] Distributed METIS partition (#2576)
* add convert.

* fix.

* add write_mag.

* fix convert_partition.py

* write data.

* use pyarrow to read.

* update write_mag.py

* fix convert_partition.py.

* load node/edge features when necessary.

* reshuffle nodes.

* write mag correctly.

* fix a bug: inner nodes in a partition might be empty.

* fix bugs.

* add verify code.

* insert reverse edges.

* fix a bug.

* add get node/edge data.

* add instructions.

* remove unnecessary argument.

* update distributed preprocessing.

* fix readme.

* fix.

* fix.

* fix.

* fix readme.

* fix doc.

* fix.

* update readme

* update doc.

* update readme.

Co-authored-by: Ubuntu <ubuntu@ip-172-31-9-132.us-west-1.compute.internal>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-2-202.us-west-1.compute.internal>
2021-02-08 17:19:18 -08:00
Jinjing Zhou 0aca566000 [Regression] Fix regression tests (#2639)
* add bench jenkins

* instance type

* fix

* fix

* fix

* 111

* test

* 111

* 111

* fix

* test

* run

* fix

* fix

* fix

* fix

* fix

* publish results

* 111

* regression

* launch ec2 script

* fix

* add

* run on master

* change

* rrr

* run gpu

* fix

* fix

* try fix

* fix

* ff

* fix

* fix

* fix

* refactor

* fix

* fix

* update

* fix

* fix

* fix

* fix

* remove import torchtext

* add shm size

* update

* fix

* fix

* fix

* fix

* fix this!!!!

* 111

* fix

* remove verbose

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* update readme

* fix

* fix

* fix

* change asv default to head

* commit sage and rgcn

* fix

* update

* add benchmarks

* add

* fix

* update

* remove RandomState

* tmp remove

* new batch

* fix

* fix

* fix

* address comment

* fix warning

* fix

* fix

* fix

* fix

* add multiupdate all

* address comment

* fix

* add benchmarks

* add

* fix timing

* fix

* push

* add -v

* [Example] NGCF (#2564)

* ngcf

* ngcf

* update

Co-authored-by: zhjwy9343 <6593865@qq.com>

* Revert "[Example] NGCF (#2564)" (#2611)

This reverts commit a75e04f408c719289f478ca129784e05655d8def.

* fix

* change task

* fix

* fix

* fix2

* enable tensoradapter when benchmark

* minor fix

* trigger ci

* fix

* fix

* fix

* fix

* fix

* fix

* lint

* fix

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
Co-authored-by: KounianhuaDu <51199171+KounianhuaDu@users.noreply.github.com>
Co-authored-by: zhjwy9343 <6593865@qq.com>
2021-02-08 17:01:35 +08:00
Quan (Andy) Gan 117dd25204 [Doc] Fix inconsistencies and GPU code (#2642)
* fix inconsistencies and GPu

* bug fixes

* fix

* trigger new tutorials
2021-02-08 16:33:39 +08:00
Tong He 8ccb8206bf [Doc] Update docs to motivate the use of set_batch_num_nodes and set_batch_num_edges (#2630)
* fix doc

* update docstring for transformation on batch graph

* update cn guide

* fix format

* update rst
2021-02-08 14:43:24 +08:00
xiang song(charlie.song) 48d357b8f6 Fix numbers (#2641)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-56-220.ec2.internal>
2021-02-08 14:05:30 +08:00
nv-dlasalle bc3a532f5e [Sampling] Implement dgl.to_block() for the GPU (#2339)
* Add start of to_block gpu implementation

* Pull in more changes from 0.4.2 cuda_to_block

* Move more code to IdArray

* Refactor DeviceNodeMapMaker

* Updates

* get compiling

* Integrate to_block

* Fix ID allocation

* Minor fixes

* Cleanup cuda calls to use cuda_common

* Reduce kernel calls

* Lint cleanup

* Expand documentation

* Remove unused function

* Rename variables for consistency

* Add doxygen comments

* Fix file extension

* Remove raw asynccopy for deviceapi

* Remove unused function

* Fix block/tile configuration

* Add cuda_device_common.cuh

* Add basic hashtable

* Migrate part of hashtable

* Refactor to use external hashtable

* Make functions members

* Format hash table functions

* Migrate duplicate filling

* Move last function over

* Refactor with cu file

* lint c++ code

* Move context check to C++ code

* Use macro switch

* Add missing files

* Update docstring

* update docs

* Move atomic functions

* Refactor hashtable

* Fix linting

* Expand docs

* Fix mismatched argument names

* Switch doxygen comments from using @param to \param

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-02-08 13:00:43 +08:00
Tianjun Xiao ae18d7f04e [Example]Add dependencies for GNN-Film and dagnn (#2633)
* add dependencies for GNN-Film and dagnn

* fix

* fix

* fix
2021-02-08 10:42:32 +08:00
Quan (Andy) Gan 98ac391ed9 [Doc] Large graph training tutorials (#2595)
* large graph training tutorials

* add full graph link prediction update

* fix

* fix

* fix

* addressed comments
2021-02-07 17:55:49 +08:00
Jinjing Zhou 0346b0aa80 [Test] Add test for Cython and force rebuild when run setup.py (#2635) 2021-02-05 22:38:46 +08:00
Tianqi Zhang (张天启) 3cf42fd199 [Bug Fix][Example] hgp-sl: More info for running (#2634)
* [Bug Fix][Example] hgp-sl: More info for running

* change list to table

* change dataset summary in gxn and sagpool

* remove redundent

Co-authored-by: Tong He <hetong007@gmail.com>
2021-02-05 18:26:33 +08:00
Mufei Li 3ca52b1e59 [Fix] Various Fix after Bug Bash (#2620)
* update

* Misc fix

* Update
2021-02-05 17:34:16 +08:00
Chen Sirui f8c9d58aed [Bug] Hardgat enable gpu training (#2632)
* add multihead in DotGatConv

* Fix spacing issue

* Add Unit test for dotgat

* Modified Unit test for dotgat

* Add transformer like divisor

* Update dotgatconv.py

* Update hgao.py

* Update train.py

Co-authored-by: Chen <chesirui@3c22fbe5458c.ant.amazon.com>
Co-authored-by: Zihao Ye <expye@outlook.com>
2021-02-05 16:52:41 +08:00
xiang song(charlie.song) 23afe911b2 [Bug Fix] Fix several sparse optimizer bugs (#2596)
* Fix pytorch TCP kvstore bug

* lint

* Fix

* upd

* Fix lint

* Fix

* trigger

* fix

Co-authored-by: Ubuntu <ubuntu@ip-172-31-56-220.ec2.internal>
2021-02-05 11:49:41 +08:00
KounianhuaDu 2f71bc507a [Example] Neural Graph Collaborative Filtering (NGCF). (#2612)
* ngcf

* ngcf

* update

* ngcf

* ngcf

* remove data

* update

* data

Co-authored-by: zhjwy9343 <6593865@qq.com>
Co-authored-by: Zihao Ye <expye@outlook.com>
2021-02-05 00:32:21 +08:00
Minjie Wang 469088ea52 [Test] Add benchmarks for end-to-end training w/ UDF message passing (#2609)
* model acc test

* add model speed test

* [Example] NGCF (#2564)

* ngcf

* ngcf

* update

Co-authored-by: zhjwy9343 <6593865@qq.com>

* Revert "[Example] NGCF (#2564)" (#2611)

This reverts commit d88a9b1b2c52becb8cc7faebceca6e52c763de19.

* use timer

Co-authored-by: KounianhuaDu <51199171+KounianhuaDu@users.noreply.github.com>
Co-authored-by: zhjwy9343 <6593865@qq.com>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Zihao Ye <expye@outlook.com>
2021-02-04 14:40:00 +08:00
张恒瑞 94ab9709a7 [Bugfix] fix a typo in consis_loss function (#2616) 2021-02-04 11:59:41 +08:00
wcyjames ff345c2e22 [Example] update PointNet and PointNet++ examples for Part Segmentation (#2547)
* [Model] update PointNet example for Part Segmentation

* Fixed issues with pointnet examples

* update the README

* Added image

* Fixed README and tensorboard arguments

* clean

* Add timing

* Update README.md

* Update README.md

Fixed a typo

Co-authored-by: Tong He <hetong007@gmail.com>
2021-02-03 22:15:46 +08:00
Mufei Li b1840f49fa [Dataset] Change the Data Type of the Node Features in GINDataset/TUDataset from Float64 to Float32 (#2592)
* Update

* Update

* Update

* update

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Tong He <hetong007@gmail.com>
2021-02-03 21:00:45 +08:00