mirrors / dmlc--dgl

提交

2,937 提交 34 分支 42 Git 标签
test_sparse_compile
提交图

2937 次代码提交

此分支
此分支
所有分支
作者 SHA1 备注 提交日期
Quan (Andy) Gan f94fa9d975 typo (#5133) 2023-01-11 22:21:11 +08:00
Hongzhi (Steve), Chen f1ed5d6de7 blabla (#5152) 
Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-01-11 20:21:55 +08:00
Hongzhi (Steve), Chen 2a78e48f4e polish (#5150) 
Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-01-11 19:55:38 +08:00
Hongzhi (Steve), Chen 41b4180950 Polish (#5149) 
Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-01-11 19:31:08 +08:00
Hongzhi (Steve), Chen 9c647d8095 [Sparse] Add elementwise mul and div method, and polish the python doc. (#5142) 
* add mul and div

* polish

* equivalent

* indent

* double

Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-01-11 19:07:32 +08:00
Chang Liu 491c19fe4a [Bugfix] Fix the call of tar.extractall() method (#5139) 2023-01-10 23:11:56 -08:00
Rhett Ying 46455328e4 [CI] fix bugs for multigpu benchmarks (#5140) 2023-01-11 14:50:37 +08:00
Hongzhi (Steve), Chen d837029925 try again. (#5141) 
Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-01-11 12:59:50 +08:00
Xin Yao e032de5848 [Bugfix] Avoid initializing CUDA context at importing (#5134) 
* Avoid initializing CUDA at importing

* renaming
 2023-01-11 12:30:20 +08:00
Hongzhi (Steve), Chen 1a33ee99d3 [Sparse] Update doc string of element-wise ops for DiagMatrix and SparseMatrix. (#5136) 
* doc

* minor

* black

* no-lint

* reorg

Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-01-11 12:08:24 +08:00
Hongzhi (Steve), Chen c5119480f2 [Sparse] Remove unnecessary API exposure. (#5135) 
* remove unnecessary exposure.

* remove_add_power

* remove_warning

* reorder

Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-01-11 11:47:14 +08:00
Quan (Andy) Gan c6228c0118 update suffix (#5128) 2023-01-10 09:56:09 +08:00
Rhett Ying b7042a141c [BugFix] fix unexpected exception when agg_type is GCN and bias disabled (#5127) 
* [BugFix] fix unexpected exception when agg_type is GCN and bias is disabled

* Update sageconv.py

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
 2023-01-10 09:05:44 +08:00
TrellixVulnTeam 5c6af80444 CVE-2007-4559 Patch (#5122) 
* Adding tarfile member sanitization to extractall()

* Update utils.py

Co-authored-by: Hongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
 2023-01-09 11:49:10 +08:00
Quan (Andy) Gan f1fb859f99 bump version to 1.0 (#5123) 2023-01-09 11:28:26 +08:00
Quan (Andy) Gan 580c702417 [DataLoader] Allow batch_size=None for GraphDataLoader (#4483) 
* overwrite default_collate_fn

* Update dataloader.py

* Update dataloader.py

* Update dataloader.py

* Update dataloader.py

* Update test_dataloader.py

* revert the test code being reverted in #4956
 2023-01-09 00:44:46 +08:00
bgawrych e296c468b2 [Optimization] Optimize bias term in SageConv layer (#4747) 
* Optimize bias term in sageconv

* fix lint

* Remove bias sharing

* Update sageconv.py

Co-authored-by: Bartlomiej Gawrych <barlomiej.gawrych@intel.com>
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
 2023-01-08 13:06:46 +08:00
peizhou001 56ffb650fe [API Deprecation]Deprecate contrib module (#5114) 2023-01-06 20:16:01 +08:00
peizhou001 436de3d120 [API Deprecation] Remove _dataloading and tgcn example (#5118) 2023-01-06 19:51:55 +08:00
Mufei Li c8bc558880 [Sparse] Remove SparseMatrix.indices and add device/dtype conversions (#5108) 
* Update

* Update

* Update

* lint

* lint

* Update

* CI

* Update diag_matrix.py

* Update sparse_matrix.py

* Update sparse_matrix.py

Co-authored-by: Ubuntu <ubuntu@ip-172-31-36-188.ap-northeast-1.compute.internal>
Co-authored-by: Hongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
 2023-01-06 17:37:13 +08:00
peizhou001 c55ab2d147 deprecate candidates in dataloader (#5117) 2023-01-06 16:18:32 +08:00
peizhou001 3a2a5031e9 [API Deprecation]Deprecate candidates in dgl.distributed (#5116) 2023-01-06 16:17:44 +08:00
peizhou001 ab0c0ec68f [API Deprecation] Deprecate candidates in convert module (#4988) (#5115) 2023-01-06 14:45:52 +08:00
Rhett Ying 46a3fc2b5f [CI] updgrade pytorch version for benchmark CI (#5102) 
* [CI] updgrade pytorch version for benchmark CI

* update build arguments

* updage

* updage

* upgrade torch to 1.13

* updage docker image

* update cmake args

* try with cu116_torch112

* update build

* update

* update

* update

* update docker image

* update

* update

* update

* update

* final update

* fix continue running

* update

* update

* update
 2023-01-06 14:35:36 +08:00
Xin Yao dde5cf5d02 [Bug] Fix NN modules crashing with non-FP32 inputs (reopen #4262) (#4829) 
* fix fp16 in nn modules

* fix lint

* remove the example

Co-authored-by: Quan Gan <coin2028@hotmail.com>
 2023-01-06 14:18:19 +08:00
Quan (Andy) Gan 9a40d2080e [Sparse] Change CMakeFiles to build DGL Sparse for multiple PyTorch versions (#5106) 
* initial commit

* Update build.sh
 2023-01-06 14:14:20 +08:00
Andrzej Kotłowski 6069f34c4f [Performance] Fix for number of threads in COOToCSR (#5017) 
Co-authored-by: Hongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
 2023-01-06 09:14:27 +08:00
Chang Liu a506ebe82b [Example][Refactor] Multi-gpu graphsage link prediction example refactor (#5091) 2023-01-05 20:59:39 +08:00
Theodore Vasiloudis 9890201d58 [Dist] Allow reading and writing single-column vector Parquet files. (#5098) 
* Allow reading and writing single-column vector Parquet files.

These files are commonly produced by Spark ML's feature processing code.

* [Dist] Only write single-column vector files for Parquet in tests.
 2023-01-05 20:37:28 +08:00
Xin Yao 7ee550f004 update cmake for cuda12 (#5048) 2023-01-05 10:47:10 +08:00
rudongyu ec3934d0d1 [Example] Sparse Graph Transformer (#5069) 
* gt example

* update

* update

* update

* update

* lint

* Update examples/sparse/graph_transformer.py

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

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
 2023-01-05 08:50:38 +08:00
Mufei Li 9169fc3541 [Sparse] Polish sparse matrix creation functions (#5093) 
* update

* Update

* Update

* Update

* Update

Co-authored-by: Ubuntu <ubuntu@ip-172-31-36-188.ap-northeast-1.compute.internal>
 2023-01-03 22:59:57 +08:00
czkkkkkk 0698e91a0e [Sparse] Use Pytorch dense computation in SpMM tests (#5096) 2023-01-03 22:26:06 +08:00
Theodore Vasiloudis 774709d399 [Dist] Add support for Parquet-formatted edges files, remove some assumptions on edge file number. (#5051) 
* [Dist] Add support for Parquet-formatted edges files, remove some assumptions on edge file number.

* [Dist] Add parquet edges option to unit tests.

Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
 2023-01-03 16:01:50 +08:00
ZhenyuLU_Heliodore cd817a1a00 [NN] Add SpatialEncoder and SpatialEncoder3d (#4991) 
* Add SpatialEncoder and SpatialEncoder3d

* Optimize the code execution efficiency

* Fixed certain problems according to Dongyu's suggestions.

* Fix an error about probability of division by zero in PathEcoder; Change certain designs in SpatialEncoder

* Fix a typo

* polish the docstring

* fix doc

* lint

Co-authored-by: Ubuntu <ubuntu@ip-172-31-14-146.ap-northeast-1.compute.internal>
Co-authored-by: rudongyu <ru_dongyu@outlook.com>
 2023-01-03 14:02:14 +08:00
Xin Yao ce378327c7 fix bf16 tests (#5089) 2022-12-29 21:36:39 +08:00
Mufei Li 37bd092587 [Sparse] Namespace migration and doc update (#5088) 
* update

* lint

* lint

* comments

Co-authored-by: Ubuntu <ubuntu@ip-172-31-36-188.ap-northeast-1.compute.internal>
 2022-12-29 20:34:29 +08:00
czkkkkkk e65fc9ff78 [Sparse] Avoid using macros in DGL headers (#5087) 2022-12-29 17:35:29 +08:00
Minjie Wang 255ad1b35b [Sparse] New print for SparseMatrix and DiagMatrix (#5085) 
* new print func

* lintrunner
 2022-12-29 15:38:10 +08:00
Minjie Wang 025e43210f [Sparse] Add tests for sparse examples; Migrate from mock_sparse to mock_sparse2 (#5073) 
* pass gcn

* example tests

* pass gcnii, hgnn, sgc, sign

* black

* test file

* add torchmetrics

* migrate appnp, c_and_s, twirls

* fix style

* minor fix

* fix hypergraphatt and twirls

* pass gat

* fix torchmetric

* lintrunner

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
 2022-12-29 08:55:25 +08:00
Mufei Li 811e35a6ef [Sparse] Change all CHECK to TORCH_CHECK (#5082) 
* Update

* CI

* Update dgl_sparse/src/utils.h

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>

Co-authored-by: Ubuntu <ubuntu@ip-172-31-36-188.ap-northeast-1.compute.internal>
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
 2022-12-28 23:26:06 +08:00
czkkkkkk 14429df661 [Sparse] Support BSpMM and BSDDMM (#5079) 
* [Sparse] Support BSpMM and BSDDMM

* Update SpMM and SDDMM error messages

* Use TORCH_CHECK

* Update error string
 2022-12-28 17:14:12 +08:00
Hongzhi (Steve), Chen bda6a816be rename (#5058) 
Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2022-12-28 13:27:52 +08:00
Minjie Wang fa5ff2fceb [Sparse] Add coalesce and has_duplicate to SparseMatrix (#5071) 
* add coalesce() and has_duplicate()

* Add duplicate check in spspmm

* black

* fix

* clang-format

* lintrunner

* minor fix

* fix
 2022-12-28 13:10:10 +08:00
Mufei Li 8c3e7830dc [Sparse] Elementwise multiplication of a SparseMatrix object by a scalar (#5076) 
Co-authored-by: Ubuntu <ubuntu@ip-172-31-36-188.ap-northeast-1.compute.internal>
 2022-12-28 09:37:44 +08:00
Mufei Li 0f1bcd9927 [Sparse] Softmax based on Reduction and BroadcastOp (#5067) 
* Update

* update

* Update

* Update

* Update

* format

* Update

* lint

* Update

* lint

* CI

* CI

* Update

* address comments

Co-authored-by: Ubuntu <ubuntu@ip-172-31-36-188.ap-northeast-1.compute.internal>
 2022-12-27 14:50:15 +08:00
Rhett Ying 311bd88a04 [Doc] update doc page for distributed partition pipeline (#5064) 
* [Doc] update doc page for distributed partition pipeline

* update
 2022-12-26 23:11:32 +08:00
czkkkkkk 3eac464a3f [Sparse] Fix the sddmm backward problem in mock_sparse. (#5070) 2022-12-26 23:10:34 +08:00
Quan (Andy) Gan 4aff59f775 [Sparse] Rename existing DGL sparse module. (#5066) 
* rename

* next time i should use lintrunner
 2022-12-26 13:55:48 +08:00
czkkkkkk 0159c3c11d [Sparse] Add SpSpMM for sparse-sparse and sparse-diag matrix multiplication. (#5050) 
* [Sparse] Add SpSpMM

* Update matmul interface

* address comments

* fix test utils to generate only coalesced matrices

* fix linter

* fix ut

* fix

* rm print

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
 2022-12-24 15:31:09 +08:00