• 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
spmm.cu 29.9 KB