mirrors / dmlc--dgl

文件历史

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

1369 次代码提交

作者 SHA1 备注 提交日期
Hongzhi (Steve), Chen d965acab7f Change size to shape in print. (#5207) 
Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-01-19 17:21:11 +08:00
Hongzhi (Steve), Chen c4ffd75223 [Misc] Polish the doc strings. (#5203) 
Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-01-19 16:37:08 +08:00
Mufei Li dffe722f5d Update (#5196) 
Co-authored-by: Ubuntu <ubuntu@ip-172-31-36-188.ap-northeast-1.compute.internal>
 2023-01-19 10:16:46 +08:00
Hongzhi (Steve), Chen 16dd9584e6 [Sparse] Polish softmax.py. (#5182) 
* polish

* [Sparse] Polish softmax.py.

* fix parameter and example

Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-01-17 14:22:10 +08:00
Hongzhi (Steve), Chen 82eb3d71a6 [Sparse] polish strings of matmul.py and sddmm.py. (#5177) 
* polish mm docstrings

* fix typo

* small fix.

* fix

Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-01-17 13:55:07 +08:00
Hongzhi (Steve), Chen 84aae0863f [Sparse] Polish reduction.py. (#5180) 
* polish

* example

* revert_example

* pylint

* revert_input

* order

* add input as parameter

* example

Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-01-17 13:34:11 +08:00
czkkkkkk f554bc4873 [Sparse] Polish the docstring of SparseMatrix.csr() and SparseMatrix.csc() (#5179) 
* [Sparse] Polish the docstring of SparseMatrix.csr() and SparseMatrix.csc()

* Update sparse_matrix.py

* Update

Co-authored-by: Hongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
 2023-01-17 13:12:48 +08:00
Hongzhi (Steve), Chen fa74ea706a [Sparse] Polish diag_matrix.py and sparse_matrix.py. (#5178) 
* polish diagonal

* polsih

* more

* more

* add paramaters-back

Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-01-17 10:46:11 +08:00
Quan (Andy) Gan 27010dbc55 [Sparse] Sparse matrix subtraction (#5164) 
* use NotImplemented

* format

* sparse sub

* address comments

* lint

* Update elementwise_op_diag.py

* ugh

Co-authored-by: Rhett Ying <85214957+Rhett-Ying@users.noreply.github.com>
 2023-01-16 11:37:42 +08:00
czkkkkkk 419fb8151b [Sparse] Use X to represent dense tensors in sddmm (#5174) 
* [Sparse] Use X to represent dense tensors in sddmm

* indent

* pylint: disable=invalid-name

Co-authored-by: Hongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
 2023-01-15 11:52:06 +08:00
Quan (Andy) Gan acc567aa08 [Sparse] Support sparse matrix dividing scalar (#5173) 
* use NotImplemented

* format

* extend to pytorch scalar

* sparse div scalar

* oops

* Apply suggestions from code review

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

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
 2023-01-13 22:54:58 +08:00
czkkkkkk b5c5c860f6 [Sparse] Refactor matmul interface. (#5162) 
* [Sparse] Refactor matmul interface.

* Update
 2023-01-13 20:08:40 +08:00
Hongzhi (Steve), Chen 9334421d2b [Sparse] Rename as_sparse to to_sparse, dense to to_dense. (#5170) 
* as_sp_to_sp

* dense

* revert_mock

* test

* revert

Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-01-13 17:03:20 +08:00
Quan (Andy) Gan 5f5db2dfc7 [Sparse] Make functions compatible with PyTorch scalar tensors (#5163) 
* use NotImplemented

* format

* extend to pytorch scalar

* reformat

* reformat

* lint
 2023-01-13 14:00:52 +08:00
czkkkkkk 1d1b08b044 [Sparse] Fix sddmm docstring (#5169) 2023-01-13 13:53:03 +08:00
Quan (Andy) Gan 8abf9d54e3 [Sparse] Use NotImplemented to let Python dispatch types (#5160) 
* use NotImplemented

* add TypeError tests

* format

* lint

* fix type

* fix docstring

* lint

* remove redundant condition
 2023-01-12 21:12:29 +08:00
czkkkkkk bff32a0986 [Sparse] Use efficient implementation for Diag @ Sparse and Sparse @ Diag. (#5147) 2023-01-12 11:53:46 +08:00
Hongzhi (Steve), Chen b743f767b6 fix (#5151) 
Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-01-12 10:13:18 +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
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
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
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
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
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
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
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
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
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
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