• 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
__init__.py 1.08 KB