1351 Commits

Author SHA1 Message Date
Quan Gan
2bdea1df91 bump version 2023-03-14 05:18:25 +00:00
Quan Gan
cc2e9933f3 update version 2023-02-15 08:32:06 +00:00
Nicolas Castet
30f4932687 [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-15 16:09:16 +08:00
Zahi Kakish
e5b548465f 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-15 16:08:53 +08:00
Krzysztof Sadowski
31e34bc6fb fix rounding issue (#5159)
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2023-02-15 16:08:37 +08:00
Quan (Andy) Gan
3b0c0cec46 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-15 16:08:20 +08:00
Rhett Ying
1956964579 [Doc] fix docstring issue (#5260) 2023-02-14 23:16:59 +08:00
Muhammed Fatih BALIN
80902cc4b9 [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-30 03:37:17 +00:00
Hongzhi (Steve), Chen
bbf1912c38 [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-28 16:27:52 +08:00
Hongzhi (Steve), Chen
0890f56685 [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-28 16:27:52 +08:00
Hongzhi (Steve), Chen
aabbcee29c 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-28 16:27:00 +08:00
Quan Gan
4c83d023e6 internal rc release 2023-01-20 09:59:11 +00: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
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
Quan (Andy) Gan
f94fa9d975 typo (#5133) 2023-01-11 22:21:11 +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
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