文件历史

提交图

1563 次代码提交

作者 SHA1 备注 提交日期
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
Xin Yao 8e73c75f4a [Fix] Fix tensor.storage() deprecation warning (#5656) 2023-05-09 13:41:53 +08:00
paoxiaode c4c9b830ae [NN] GatedGCNConv (#5654) 2023-05-06 18:20:06 +08:00
鄢振宇Michael Yan d6c12f0714 [NN] Fix #5642 (#5652) 2023-05-06 13:03:20 +08:00
Tingyu Wang ea706cae58 [Model] Update cugraph-ops models for 23.04 release (#5540)
Co-authored-by: schmidt-ju <79865721+schmidt-ju@users.noreply.github.com>
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-36-188.ap-northeast-1.compute.internal>
Co-authored-by: peizhou001 <110809584+peizhou001@users.noreply.github.com>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-16-19.ap-northeast-1.compute.internal>
2023-05-04 17:11:22 +08:00
Theodore Vasiloudis 1ab0170a10 [Distributed] Ensure round-robin edge file downloads, reduce logging, other improvements. (#5578)
Co-authored-by: Rhett Ying <85214957+Rhett-Ying@users.noreply.github.com>
2023-04-27 11:17:55 -07: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
Hongyi ZHANG e6226e826d [LOG] check whether cpu_affinity is avaible on the platform (#5572)
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2023-04-20 15:44:52 +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
Chang Liu 62b5f50a5c [Feature] Import PyTorch's pin_memory() method for DGL graph structure (#5366) 2023-04-10 19:21:32 -07: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
Ilia Taraban acb4eb7ecd [Feature] Add bfloat16 support for CPU (#5497)
Co-authored-by: Hongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
2023-04-06 11:16:41 +02:00
DominikaJedynak 29e6661523 [Optimization] Optimize bias term in GatConv layer (#5466) 2023-04-05 23:17:10 +08:00
Mufei Li a244de579c [Dataset] WisconsinDataset (#5520)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-36-188.ap-northeast-1.compute.internal>
2023-04-04 16:44:26 +08:00
Mufei Li dd4c74ff32 [Dataset] CornellDataset and TexasDataset (#5513)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-36-188.ap-northeast-1.compute.internal>
2023-04-03 16:16:37 +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
Mufei Li 2b5921e759 [Dataset] Actor (#5511)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-36-188.ap-northeast-1.compute.internal>
2023-03-31 10:28:50 +08:00
Hongzhi (Steve), Chen 8f9f2e2a4d [Misc] Rename number_of_edges and number_of_nodes to num_edges and num_nodes. (#5472)
* core

* revert

---------

Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
2023-03-29 08:57:47 +08:00
Mufei Li 690f37bbe9 update (#5507)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-36-188.ap-northeast-1.compute.internal>
2023-03-28 19:22:44 +08:00
Hongzhi (Steve), Chen 89b321b847 [Misc] Rename number_of_edges and number_of_nodes to num_edges and num_nodes. (#5488)
* nn-only

* data-only

---------

Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
2023-03-28 12:58:54 +08:00
Hongzhi (Steve), Chen 7f5da69709 [Misc] Rename number_of_edges and number_of_nodes to num_edges and num_nodes in dist related python files. (#5489)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
2023-03-27 15:55:37 +08:00
Mufei Li 9e532e7def [Dataset] Chameleon (#5477)
* update

* update

* update

* lint

* update

* CI

* lint

* update doc

---------

Co-authored-by: Ubuntu <ubuntu@ip-172-31-36-188.ap-northeast-1.compute.internal>
2023-03-23 22:34:00 +08:00
Xin Yao 170203ae2b [Performance] Creating out buffers for segment_mm|sddmm via torch.empty() (#5462)
* update for segmentMM

* update for sddmm

* fix a bug
2023-03-23 17:41:07 +08:00
Hongzhi (Steve), Chen 0ac9bf34b5 Rename number_of_nodes, number_of_edges to num_nodes, num_edges. (#5486)
* Add num_nodes/num_edges

* additional

* Update python/dgl/graph_index.py

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

* Apply suggestions from code review

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-23 17:30:45 +08:00
ZhenyuLU_Heliodore 4534650f5f [Dataset] Add ZINC Dataset (#5428)
* Update dgl.data.rst

* Add files via upload

* Add files via upload

* Add files via upload

* Update zinc.py

* Update dgl.data.rst

* Update test_data.py

* Add files via upload

* Update cluster.py

* Update pattern.py

* Update zinc.py

* Update zinc.py

* Update test_data.py

* lint

* fix

* fix path

* update test on label shape

---------

Co-authored-by: rudongyu <ru_dongyu@outlook.com>
2023-03-22 18:04:41 +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
Mufei Li f5ddb11448 Update neighbor_sampler.py (#5450) 2023-03-14 15:46:40 +08:00
Rhett Ying d2ed57e94b [DistDGL] add log when load graph and data (#5350)
* [DistDGL] add log when load graph and data

* fix lint

* refine log

---------

Co-authored-by: Israt Nisa <neesha295@gmail.com>
2023-03-10 14:21:32 +08:00
Gleb Bazhenov 54b4bd0acd [Data] new functional for creating data splits in graph (#5418)
* new functional for creating data splits in graph

* minor fix in data split implementation

* apply suggestions from code review

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

* refactoring + unit tests

* fix test file name

* move imports to the top

* Revert "fix test file name"

This reverts commit 126323e38c8de6fdc73c826dccc9048aed8f9634.

* remove nccl submodule

* address linter issues

---------

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
Co-authored-by: Hongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
2023-03-09 18:55:25 +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
Guangyu Zhou 178290249e [Dataset] Add PATTERN dataset (#5422)
* add PATTERN dataset

* fix bug

* fix bugs

* fix issues

* refine according to dongyu's comments

---------

Co-authored-by: BuptTab <gyzhou2000@gmail.com>
Co-authored-by: rudongyu <ru_dongyu@outlook.com>
2023-03-08 14:08:12 +08:00
Hongzhi (Steve), Chen a454734f16 [Misc] Follow up on the missing comment in PR#4989. (#5437)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
Co-authored-by: Xin Yao <yaox12@outlook.com>
2023-03-08 13:32:21 +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
Tingyu Wang bfd411d0d5 [Model] Add dgl.nn.CuGraphGATConv model (#5168)
* add CuGraphGATConv model

* lintrunner

* update model to reflect changes in make_mfg_csr(), move max_in_degree to forward()

* simplify pytest markers

* fall back to FG option for large fanout

* update error msg

* add feat_drop and activation options

* add residual option

* Update python/dgl/nn/pytorch/conv/cugraph_gatconv.py

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

* Update python/dgl/nn/pytorch/conv/cugraph_gatconv.py

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

* reset res_fc

---------

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2023-03-04 14:10:06 +08:00
Mufei Li f00cd6efbd [Utils] Edge and LINKX homophily measure (#5382)
* Update

* lint

* lint

* r prefix

* CI

* lint

* skip TF

* Update

* edge homophily

* linkx homophily

* format

* skip TF

* fix test

* update

* lint

* lint

* review

* lint

* update

* lint

* update

* CI

---------

Co-authored-by: Ubuntu <ubuntu@ip-172-31-36-188.ap-northeast-1.compute.internal>
2023-03-03 16:58:20 +08:00
Zhiteng Li a53ecd229b [Dataset] Add CLUSTER dataset (#5389)
* add CLUSTER dataset

* refine according to dongyu's comments

---------

Co-authored-by: rudongyu <ru_dongyu@outlook.com>
2023-03-01 16:36:12 +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
Mufei Li fcf5ad5f49 [Utils] Node homophily measure (#5376)
* Update

* lint

* lint

* r prefix

* CI

* lint

* skip TF

* Update

---------

Co-authored-by: Ubuntu <ubuntu@ip-172-31-36-188.ap-northeast-1.compute.internal>
2023-02-24 14:26:02 +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
peizhou001 7ff041528c [Refactor] Add default ffi namespce capi (#5359) 2023-02-22 21:33:30 +08:00