mirrors / dmlc--dgl

文件历史

3,326 提交 34 分支 42 Git 标签
peizhou/negative__patch
提交图

1563 次代码提交

作者 SHA1 备注 提交日期
Tingyu Wang bcf9923b32 [Model] Add dgl.nn.CuGraphSAGEConv model (#5137) 
* add CuGraphSAGEConv model

* fix lint issues

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

* lintrunner

* allow reset_parameters()

* remove norm option, simplify test

* allow full graph fallback option, add example

* address comments

* address reviews

---------

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
 2023-02-22 13:43:28 +08:00
Quan (Andy) Gan d8ca6317a3 [Release] Bump nightly version (#5357) 
* bump version

* Update update_version.py
 2023-02-22 09:13:38 +08:00
Hongzhi (Steve), Chen 9ce80e858b [Misc] Autoformat python dgl. (#5335) 
* autofix

* sort

* sort

---------

Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
 2023-02-21 23:38:30 +08:00
Hongzhi (Steve), Chen 529b26627b autofix (#5337) 
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
 2023-02-21 20:12:43 +08:00
Hongzhi (Steve), Chen 4706b6bf72 [Misc] Autoformat python dgl. (#5338) 
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
 2023-02-21 15:24:06 +08:00
Hongzhi (Steve), Chen a566b60be4 auto-fix (#5331) 
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
 2023-02-21 01:25:16 +08:00
Hongzhi (Steve), Chen d18274885d autofix2 (#5333) 
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
 2023-02-20 23:27:27 +08:00
Hongzhi (Steve), Chen 3e5137feaf [Misc] Autoformat python dgl (Part 2). (#5332) 
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
 2023-02-20 20:56:25 +08:00
Hongzhi (Steve), Chen 6e58f5f1a5 [Misc] Auto format with manual fix of 3 files and add pylint: disable= too-many-lines for functional.py. (#5330) 
* blabal

* 2more

---------

Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
 2023-02-20 17:14:10 +08:00
Hongzhi (Steve), Chen ab812179a3 [Misc] Auto reformat data/. (#5318) 
* data

* lintrunner

---------

Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
 2023-02-19 22:50:55 +08:00
Hongzhi (Steve), Chen 63ac788ff4 auto-reformat-nn (#5319) 
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
 2023-02-19 08:33:30 +08:00
Hongzhi (Steve), Chen 0b3a447ba4 auto format distributed (#5317) 
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
 2023-02-19 08:33:09 +08:00
Tingyu Wang 19b3cea166 [Model] Update CuGraphRelGraphConv to use pylibcugraphops=23.02 (#5217) 
* update cugraph_relgraphconv

* update equality test

* update cugraph rgcn example

* update RelGraphConvAgg based on latest API changes

* enable fallback option to fg when fanout is large

---------

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
 2023-02-15 20:11:20 +08:00
Krzysztof Sadowski f5afc6eaa2 fix rounding issue (#5159) 
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
 2023-02-13 18:02:55 +08:00
Zahi Kakish 21733e903b Documentation equation syntax fix for EGATConv (#5284) 
Some weird formatting occurred in the documentation for EGATConv, so this quick commit should fix that.

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
 2023-02-13 14:58:11 +08:00
Quan (Andy) Gan f62669b05c enable sparse on windows and mac (#5277) 
* enable sparse on windows and mac

* that was stupid

* let's see what's going on..

* [Sparse] Fix the import error on Mac OS.

When using template functions that are defined in source files from DGL,
the loader of MacOS somehow cannot find their definitions. This fix simply
avoids depending on template functions from DGL headers.

With this fix, the sparse tests all pass on the MAC environment.

* ok this is the problem

* make errors clearer

* uh

* test

* Update __init__.py

* disabling ddp on windows

---------

Co-authored-by: czkkkkkk <zekucai@gmail.com>
 2023-02-13 13:16:17 +08:00
Nicolas Castet af79e54049 [Bugfix] Fix Column.record_stream(...) for unmaterialized tensors (#5240) 
* Fix Column.record_stream(...) for unmaterialized tensors

* lint

* Record streams for indices on non-materialized columns

* add docstring

* fix for cpu index

* Record also stream on storage

* Always call self.storage.record_stream when storage is on GPU

* fix lint

---------

Co-authored-by: Xin Yao <xiny@nvidia.com>
 2023-02-10 14:36:46 +08:00
ZhenyuLU_Heliodore 5598503aa0 [Transform] Add SVDPE Transform Module (#5121) 
* add SVD positional encoding

* modify importing module

* Fixed certain problems

* Change the test unit to a nonsigular one

* Fixed typo and make accord with lintrunner

* added svd_pe into dgl.rst

* Modified dgl.rst
 2023-02-08 12:29:03 +08:00
Rhett Ying cf67829998 [Doc] fix docstring issue (#5260) 2023-02-01 16:13:43 +08:00
Hongzhi (Steve), Chen b3308ae795 Cleanup mock_sparse related code. (#5230) 
* remove_mock_sparse_example

* mock_sparse_test

* remove_mock_sparse

---------

Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-02-01 15:17:14 +08:00
Muhammed Fatih BALIN 5888d9d1b6 [Doc] Labor sampling improvement (#5221) 
* [Doc] Labor sampling improvement

When importance sampling option is used, the returned blocks include edge weights that need to be used during the message passing operation. So, I am including this in the documentation.

* Add documentation to weighted sampling case also
 2023-01-29 21:32:09 +08:00
Hongzhi (Steve), Chen 4b5fa83bcd [Sparse] Swap examples of from_coo and spmatrix. (#5228) 
* swap-example

* size-shape

Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-01-27 10:51:57 +08:00
Hongzhi (Steve), Chen 387ae76df3 [Doc] Update with Mono font to keep doc string consistent. (#5213) 
* part1

* more

* more

* spmatrix

* code

Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-01-23 09:48:07 +08:00
Hongzhi (Steve), Chen d61012e094 Change example code for indices creation from two line to one line to avoid test break. (#5222) 
* add_backslash

* fix

* more

* blabla

Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-01-23 02:21:57 +08:00
Hongzhi (Steve), Chen fbbe6d61c9 [Doc] Update all canonical creation example from from_coo to spmatrix. (#5208) 
* fix

* sddmm

* change

* blabla

* blabla

* revert

* newpr

* tocuda

Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-01-20 17:44:07 +08:00
Quan (Andy) Gan abfc4c31f5 [Sparse] Clarification on dimensions for sparse reduction and softmax docstrings (#5197) 
* fix sparse reduce and softmax doc

* Update softmax.py
 2023-01-20 14:31:33 +08:00
Hongzhi (Steve), Chen 7a2dd5ae96 32 (#5216) 
Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-01-20 11:26:26 +08:00
Hongzhi (Steve), Chen c89ee05f2d examples (#5214) 
Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2023-01-20 10:46:12 +08:00
czkkkkkk 20ce9be2c6 [Sparse] Add spmatrix creation function (#5209) 
* [Sparse] Add spmatrix creation function

* Update

* Update sparse_matrix.py

* Update sparse_matrix.py

Co-authored-by: Hongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
 2023-01-20 09:49:49 +08:00
czkkkkkk d1e5700676 [Sparse] Polish the docstring of sparse_matrix.py (#5202) 
* [Sparse] Polish the docstring of sparse_matrix.py

* Update

* Update

* Update

* Update
 2023-01-20 00:49:20 +08:00
Quan (Andy) Gan f145097f9a [Release] Conda/CMake changes for 1.0 release (#5186) 
* update conda branch

* fix conda

* update conda

* update

* update windows build script

* copy static library to build dir

* try again

* fix

* lint
 2023-01-19 22:28:04 +08:00
czkkkkkk db77283130 [Sparse] Only support scalar or vector values in SparseMatrix creators (#5204) 
* [Sparse] Only support scalar or vector values in SparseMatrix creators

* Update

* Update sparse_matrix.py

Co-authored-by: Hongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
 2023-01-19 21:46:46 +08:00
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
Quan (Andy) Gan 84e4d0215f change regex (#5158) 2023-01-12 15:04:22 +08:00
czkkkkkk bff32a0986 [Sparse] Use efficient implementation for Diag @ Sparse and Sparse @ Diag. (#5147) 2023-01-12 11:53:46 +08:00