 Israt Nisa
|
188152b853
|
[Feature] Add Heterograph support on Python for builtin unary msg functions (copy_u, copy_e) (#2989)
* heterograph for binary func
* Added SDDMM support
* Added unittest
* added binary test cases
* unary mfuncs works
* Fixed lint err
* lint check and others
* link check
* fixed import *_hetero issue
* lint check
* replace torch with dgl backend
* lint cehck
* removed torch from test
* skip mxnet unittest
* skip gpu test
* Remove unused/duplicated code
* minor
* changed data structure of ndata and edata
* link check
* reorganized
* minor lint
* minor lint
* raise error for udf func
* lint check
* fix for CUDA 10.1
* add a note for future cross-type max/min reducing
* Add support CUDA < 11
* lint check
* tidied C code
* remove dummy GSDDMM_hetero backward implementation
Co-authored-by: Israt Nisa <nisisrat@amazon.com>
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
Co-authored-by: Quan Gan <coin2028@hotmail.com>
|
2021-07-06 20:41:58 +08:00 |
|
|
Konstantin Sofeikov
|
e22087aa36
|
[BUG] fix for zero degree nodes (#2786)
* fix for zero degree nodes
* Update python/dgl/ops/spmm.py
linting fix
Co-authored-by: Zihao Ye <zihaoye.cs@gmail.com>
Co-authored-by: Konstantin Sofeikov <konstantin.sofeikov@complyadvantage.com>
Co-authored-by: Zihao Ye <expye@outlook.com>
Co-authored-by: Zihao Ye <zihaoye.cs@gmail.com>
|
2021-03-25 22:09:53 +08:00 |
|
|
Zihao Ye
|
2fc280ba65
|
only filling zeros for min/max reducer (#2026)
|
2020-08-15 12:43:36 +08:00 |
|
|
Quan (Andy) Gan
|
63e2ba23c6
|
[WIP][Kernel] Set the built-in reduce result of zero-degree nodes to 0 in C (#2017)
* test idea
* cuda kernels
* lint and fixes
* lint
* change to another strategy
* use infinity
* fix
Co-authored-by: Zihao Ye <expye@outlook.com>
|
2020-08-14 22:59:24 +08:00 |
|
|
Zihao Ye
|
6294677f8a
|
[hotfix] Set reduce results to all zero for nodes with zero in-degrees. (#2011)
|
2020-08-13 01:20:56 +08:00 |
|
|
Minjie Wang
|
22167f7203
|
[Refactor] Enable new kernel in all message passing APIs (#1953)
* WIP: frame refactor
* new frame
* simple update_all builtin
* move all subgraph routines into the same file
* sddmm & spmm schedule; node & edge udf
* degree bucketing
* some tricky 0deg corner cases
* bug in frame append
* merge test_hetero_basics and test_basics
* some code rearange
* fix test_heterograph
* add mean spmm
* enable all builtin combinations
* pass gpu test
* pass pytorch tests
* wip
* fix some pt debugging codes
* fix bug in mxnet backward
* pass all mxnet utests
* passed tf tests
* docstring
* lint
* lint
* fix broadcasting bugs
* add warning and clamp for mean reducer
* add test for zero-degree mean
* address comments
* lint
* small fix
|
2020-08-07 15:40:25 +08:00 |
|
|
Zihao Ye
|
a28bfa9f5f
|
[hotfix] Activate kernel unittest for tensorflow (#1895)
* upd
* upd
* upd
* upd
* upd
* trigger
* simplify unittest
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
|
2020-07-31 19:57:53 +08:00 |
|
|
Zihao Ye
|
05a4337901
|
[Doc] Introduce the relation of message passing APIs and operators in docstring. (#1878)
* upd
* fix typo
* upd
|
2020-07-29 00:41:47 +08:00 |
|
|
Zihao Ye
|
2fa2b4534e
|
[Feature] Support higher order derivative for message passing. (#1877)
* upd
* fix typo
|
2020-07-28 23:11:06 +08:00 |
|
|
Zihao Ye
|
2b8eb5be6f
|
[refactor] Move spmm and sddmm operators to operator module (#1873)
* upd
* upd
* upd
* upd
|
2020-07-28 18:39:12 +08:00 |
|