 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 |
|
|
Minjie Wang
|
8a07ab7737
|
[Doc] Tutorials re-organization (#2683)
* reorg
* change titles
* rm some stale API doc; minor fix
* fix docs
* add warning
* rm new-tutorial run in ci
* lint
|
2021-02-20 17:42:45 +08:00 |
|
|
Zihao Ye
|
0f56575938
|
[feature] Allow copy_e + udf reduce function (#1996)
* up
* pylint
|
2020-08-11 18:47:40 +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 |
|