1. 03 Dec, 2021 1 commit
    • Israt Nisa's avatar
      [Feature] Add Min/max reducer in heterogeneous API for unary message functions (#3514) · cb0e1103
      Israt Nisa authored
      
      
      * min/max support for forward CPU heterograph
      
      * Added etype with each argU values
      
      * scatter_add needs fix
      
      * added scatter_add_hetero. Grads dont match for max reducer
      
      * storing ntype in argX
      
      * fixing scatter_add_hetero
      
      * hetero matches with torch's scatter add
      
      * works copy_e forward+cpu
      
      * added backward for copy_rhs
      
      * Computes gradient for all node types in one kernel
      
      * bug fix
      
      * unnitest for max/min on CPU
      
      * renamed scatter_add_hetero to update_grad_minmax_hetero
      
      * lint check and comment out cuda call for max. Code is for CPU only
      
      * lint check
      
      * replace inf with zero
      
      * minor
      
      * lint check
      
      * removed LIBXSMM code from hetro code
      
      * fixing backward operator of UpdateGradMinMaxHetero
      
      * removed backward from update_grad_minmax_hetero
      
      * docstring
      
      * improved docstring and coding style
      
      * Added pass by pointer for output
      
      * typos and pass by references
      
      * Support for copy_rhs
      
      * Added header <string>
      
      * fix bug in copy_u_max
      
      * Added comments and dimension check of all etypes
      
      * skip mxnet check
      
      * pass by pointer output arrays
      
      * updated docstring
      Co-authored-by: default avatarIsrat Nisa <nisisrat@amazon.com>
      Co-authored-by: default avatarQuan (Andy) Gan <coin2028@hotmail.com>
      cb0e1103
  2. 11 Oct, 2021 1 commit
  3. 07 Sep, 2021 1 commit
  4. 06 Jul, 2021 1 commit
    • Israt Nisa's avatar
      [Feature] Add Heterograph support on Python for builtin unary msg functions... · 188152b8
      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: default avatarIsrat Nisa <nisisrat@amazon.com>
      Co-authored-by: default avatarMinjie Wang <wmjlyjemaine@gmail.com>
      Co-authored-by: default avatarQuan Gan <coin2028@hotmail.com>
      188152b8
  5. 25 Mar, 2021 1 commit
  6. 15 Aug, 2020 1 commit
  7. 14 Aug, 2020 1 commit
  8. 12 Aug, 2020 1 commit
  9. 07 Aug, 2020 1 commit
    • Minjie Wang's avatar
      [Refactor] Enable new kernel in all message passing APIs (#1953) · 22167f72
      Minjie Wang authored
      * 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
      22167f72
  10. 31 Jul, 2020 1 commit
  11. 28 Jul, 2020 3 commits