mirrors / dmlc--dgl

项目文件夹

dmlc--dgl
文件
0.7.x
T
文件历史
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
..
__init__.py
[Performance] Use segment operators for graph readout. (#2361)
2020-11-23 01:45:33 +08:00
edge_softmax.py
[Doc] Tutorials re-organization (#2683)
2021-02-20 17:42:45 +08:00
sddmm.py
[Feature] Add Heterograph support on Python for builtin unary msg functions (copy_u, copy_e) (#2989)
2021-07-06 20:41:58 +08:00
segment.py
[doc] Add docstring for segment reduce. (#2375)
2020-11-27 18:04:13 +08:00
spmm.py
[Feature] Add Heterograph support on Python for builtin unary msg functions (copy_u, copy_e) (#2989)
2021-07-06 20:41:58 +08:00