 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 |
|