mirrors / dmlc--dgl

文件历史

3,215 提交 34 分支 42 Git 标签
fix-gpu-sampling-doc
提交图

1492 次代码提交

作者 SHA1 备注 提交日期
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
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
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