mirrors / dmlc--dgl

文件历史

3,215 提交 34 分支 42 Git 标签
fix-gpu-sampling-doc
提交图

90 次代码提交

作者 SHA1 备注 提交日期
peizhou001 c9c165f7cb [GraphBolt] Add fanout for neighbor sampling (#5768) 
[Graphbolt] Add fanout for sampling
 2023-06-02 18:50:04 +08:00
peizhou001 a99095e78c [GraphBolt]Simple neighbor sampling that picking all neighbors for given nodes (#5765) 
[GraphBolt] Add neighbor sampling picking all neighbors for given nodes
 2023-06-02 16:05:05 +08:00
peizhou001 ee00729bbe [Graphbolt] Define API for Neighbor sampling (#5766) 
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-25-242.ap-northeast-1.compute.internal>
 2023-06-02 14:17:36 +08:00
czkkkkkk dc8a027903 [Graphbolt] Enable CSCSamplingGraph copy to/load from shared memory and add unittests in Python (#5742) 2023-05-30 19:05:00 +08:00
Rhett Ying 00972deecf [GraphBolt] convert dgl partition to CSCSamplingGraph (#5736) 2023-05-29 16:49:39 +08:00
Rhett Ying 236ffa0f63 [GraphBolt] add in_subgraph() API at the python level (#5743) 2023-05-26 15:37:07 +08:00
Chang Liu 68275b9f3e [Feature] Decompose dataset to each process to add multi-node support for dataloader (#5617) 2023-05-24 16:30:56 -07:00
Rhett Ying 70af8f0d85 [GraphBolt] add support for load/save CSCSamplingGraph in python level (#5733) 2023-05-24 17:24:43 +08:00
peizhou001 567a9df2d0 [GraphBolt] Add csc sampling graph python side code (#5700) 2023-05-23 16:59:59 +08:00
peizhou001 ce66280d9d [GraphBolt] Reorganzie graphbolt module (#5711) 
Co-authored-by: Ubuntu <ubuntu@ip-172-31-25-242.ap-northeast-1.compute.internal>
 2023-05-19 16:55:03 +08:00
Rhett Ying 0b386a1d4f Merge Graphbolt to master (#5680) 
Co-authored-by: Ubuntu <ubuntu@ip-172-31-25-242.ap-northeast-1.compute.internal>
 2023-05-12 12:09:17 +08:00
Kunal Mukherjee 484fabdcc1 [Model] Implemented PGExplainer for Homogeneous graph (#5550) 
Co-authored-by: kxm180046 <kxm180046@utdallas.edu>
Co-authored-by: Infinity_X <Infinity_X@sinomo.net>
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
 2023-05-10 20:31:48 +08:00
paoxiaode c4c9b830ae [NN] GatedGCNConv (#5654) 2023-05-06 18:20:06 +08:00
Kunal Mukherjee c0f346e754 [Model] Implemented SubgraphX Explainer for Heterogeneous graph (#5530) 
Co-authored-by: kxm180046 <kxm180046@utdallas.edu>
Co-authored-by: Nick Baker <ndbaker1@outlook.com>
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
 2023-04-21 14:05:02 +08:00
czkkkkkk f41934df87 [Sparse] Add conversion between DGLGraph and SparseMatrix. (#5553) 2023-04-20 18:00:34 +08:00
Zhiteng Li b58416f76a [NN] Refine the API of GraphormerLayer (#5565) 
Co-authored-by: rudongyu <ru_dongyu@outlook.com>
 2023-04-19 16:36:25 +08:00
czkkkkkk b57c56d9a1 [Sparse] Support broadcasting operators (#5544) 2023-04-15 10:36:12 +08:00
czkkkkkk 7c465d20fc [Sparse] Support spspdiv (#5541) 
Co-authored-by: Hongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
 2023-04-14 16:46:51 +08:00
rudongyu bb1f8850e5 [NN] Refactor the Code Structure of GT (#5100) 2023-04-14 11:50:14 +08:00
Venzino.Han f1b0a07984 [NN] add egatconv edge_weight (#5539) 
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
 2023-04-13 20:54:42 +08:00
czkkkkkk 893098888e [Sparse] Support spspmul (#5543) 2023-04-13 09:07:49 +08:00
peizhou001 4864a9f903 [Bug] Merge rather replace when find reverse eids (#5532) 
Co-authored-by: Ubuntu <ubuntu@ip-172-31-16-19.ap-northeast-1.compute.internal>
 2023-04-12 10:42:42 +08:00
Tianqi Zhang (张天启) 27b008b9b1 [BugFix] Fix unstable behavior of beuteforce-sharemem KNN (#5515) 2023-04-10 22:26:10 +08:00
schmidt-ju 97286f98d3 [NN] Add EdgeGAT operator (#5282) 
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
 2023-04-09 22:54:49 +08:00
czkkkkkk 4463b3d6a5 [Sparse] Support strided tensor when calling old DGL APIs (#5506) 2023-03-31 23:18:55 +08:00
Venzino.Han d45eafd47f [NN] add gatconv edge_weight (#5503) 
Co-authored-by: venzino-han <venzino.han@buzzvil.com>
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
 2023-03-31 19:53:49 +08:00
Hongzhi (Steve), Chen 5b409bf7f5 Rename_test (#5487) 
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
 2023-03-24 18:44:24 +08:00
czkkkkkk 866c70dace [Sparse] Fix bugs in Diag format conversions (#5465) 2023-03-21 09:24:53 +08:00
Hongzhi (Steve), Chen 8a83027274 [Misc] Merge test utils. (#5440) 
* merge

* format

* rename

* sort

* sort

* update

* update

* update

* Update tests/utils/checks.py

Co-authored-by: Mufei Li <mufeili1996@gmail.com>

---------

Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
 2023-03-20 15:32:45 +08:00
czkkkkkk 92e383c3b8 [Sparse] Remove the Python DiagMatrix class (#5444) 
* [Sparse] Remove the Python DiagMatrix class

* Update

* Update

* update

* Update
 2023-03-15 18:17:52 +08:00
Hongzhi (Steve), Chen e250ba3d99 [Misc] Rename test_utils -> pytests_utils. (#5438) 
* rename

* sort

* rename

---------

Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
 2023-03-09 09:42:29 +08:00
kkranen fe3d29ac4d [Dataloading] Ignore edge types without reverse types in edge dataloader (#5411) 
* Bugfix for reverse edge issue.

* Resolved copy/paste transcription error

* lint

* Added unit tests for graph with not all reverses.

* linting + added backend device test

---------

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
Co-authored-by: Rhett Ying <85214957+Rhett-Ying@users.noreply.github.com>
 2023-03-08 22:04:51 +08:00
Xin Yao ae4a5b7394 [Feature] Add state_dict, load_state_dict, param_groups to dgl.optim.SparseGradOptimizer (#5311) 
* init update

* all get/set optm_state

* add unit tests

* add docstring

* fix for multiple embeddings

* move embedding methods to private

* fix lint

* fix unit tests

* resolve comments

* merge master
 2023-03-08 16:46:34 +08:00
Xin Yao 8d5d8962ad [Refactor] Replace third_party/nccl with PyTorch's NCCL backend (#4989) 
* expose GeneratePermutation

* add sparse_all_to_all_push

* add sparse_all_to_all_pull

* add unit test

* handle world_size=1

* remove python nccl wrapper

* remove the nccl dependency

* use pinned memory to speedup D2H copy

* fix lint

* resolve comments

* fix lint

* fix ut

* resolve comments
 2023-03-08 12:59:10 +08:00
Quan (Andy) Gan 851d66fa65 [Bugfix] Fix duplicate worker_init_fn argument when provided in DataLoader (#5420) 
* fix duplicate worker_init_fn

* lint

* lint again

* uugh
 2023-03-06 23:41:53 +08:00
czkkkkkk 999c624582 [Sparse] Support converson to/from torch sparse tensor. (#5388) 
* [Sparse] Support converson to/from torch sparse tensor.

* Update
 2023-02-28 17:27:23 +08:00
czkkkkkk 5ffd2a022c [Sparse] Support column-wise softmax (#5377) 
* [Sparse] Support column-wise softmax

* Update python/dgl/sparse/softmax.py

Co-authored-by: Mufei Li <mufeili1996@gmail.com>

---------

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
 2023-02-24 10:03:43 +08:00
Kunal Mukherjee 45153fc078 [Model] Implemented SubgraphX Explainer for Homogeneous graph (#5315) 
* subgraphx commit

* nits

* newline eof added

* lint fix

* test script updated to use default values

* lint fix

* graphs that are used for test cases are updated to a small graph

* lint formatted

* test paramter adj to complete the test under 20s

* lint fixes

---------

Co-authored-by: kxm180046 <kxm180046@utdallas.edu>
 2023-02-23 12:50:31 +08:00
Hongzhi (Steve), Chen 74c9d27d16 [Misc] Auto-format tests. (#5313) 
* [Misc] Auto-format tests.

* more

---------

Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
 2023-02-17 23:33:44 +08:00
Hongzhi (Steve), Chen 86193c26f7 [Misc] Reorg python test (with backend) according to the guideline. (#5312) 
* rename

* remove test transpose

* udpate

* rename

* cleanup

---------

Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
 2023-02-17 16:11:39 +08:00