mirrors / dmlc--dgl

文件历史

3,142 提交 34 分支 42 Git 标签
1.1.x
提交图

37 次代码提交

作者 SHA1 备注 提交日期
Andrei Ivanov 3e04a6715a Removing warnings appearing in heterograph tests (#6095) 
Co-authored-by: Xin Yao <xiny@nvidia.com>
 2023-08-10 01:58:49 +00:00
Jing Zhu 9e33ccde45 [SpotTarget]Edge sampler with excluding edges adjacent to low-degree nodes (#5893) 
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
Co-authored-by: ZYH <31127200+Tonyzhou98@users.noreply.github.com>
Co-authored-by: Hongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
 2023-08-10 01:58:49 +00:00
Zhiteng Li c7c6cdf407 [NN] Add EGT Layer (#6056) 
Co-authored-by: rudongyu <ru_dongyu@outlook.com>
 2023-08-10 01:58:48 +00:00
nv-dlasalle 3936bb7a7e [Bugfix] Allow pinning of empty tensor (#6044) 2023-08-10 01:58:48 +00:00
Muhammed Fatih BALIN 39b0a52831 [Performance][CUDA] Labor UVA optimization (#5885) 
Co-authored-by: Xin Yao <xiny@nvidia.com>
Co-authored-by: Rhett Ying <85214957+Rhett-Ying@users.noreply.github.com>
 2023-08-10 01:58:48 +00:00
Andrzej Kotłowski 62a2430e74 [Test] Run test_spmm also on floats (#5930) 2023-08-10 01:58:47 +00:00
rudongyu 5a310c4689 [NN] Refactor SpatialEncoder3d (#5894) 2023-08-10 01:58:47 +00:00
Andrzej Kotłowski b413bf27c3 [TESTS] Make pytorch test_nn.py fully reproducible (#5887) 2023-08-10 01:58:47 +00:00
Zhiteng Li 051258f537 [NN] Refactor DegreeEncoder, SpatialEncoder and PathEncoder (#5799) 
Co-authored-by: rudongyu <ru_dongyu@outlook.com>
 2023-06-20 07:47:55 +00:00
Chang Liu 53f2c30942 [Feature] Decompose dataset to each process to add multi-node support for dataloader (#5617) 2023-06-20 07:43:05 +00: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