- 07 Sep, 2021 1 commit
-
-
Israt Nisa authored
* Added binary builtinMsgFunc forward() for heterograph * Added backward for u_op_v * Supports all binary builtin forward * Supports binary message funcs with reduce func sum * lint check * removed import torch from unittest * enabled GPU test * lint check * Fixed docstrings * rename func get_hs_id * edited comment Co-authored-by:Israt Nisa <nisisrat@amazon.com>
-
- 06 Jul, 2021 1 commit
-
-
Israt Nisa authored
[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>
-