• xiang song(charlie.song)'s avatar
    [New Feature] Per edge type sampler for to_homogeneous graphs. (#3131) · ba7e7cf9
    xiang song(charlie.song) authored
    
    
    * fix.
    
    * fix.
    
    * fix.
    
    * fix.
    
    * Fix test
    
    * Deprecate old DistEmbedding impl, use synchronized embedding impl
    
    * Basic imple of heterogeneous on homogenenous sampling
    
    * make pass
    
    * Pass C++ test
    
    * Add python test code
    
    * lint
    
    * lint
    
    * Add MultiLayerEtypeNeighborSampler
    
    * Add unitest for single machine dataloader
    
    * Add dist dataloader test for edge type sampler
    
    * Fix lint
    
    * fix
    
    * support for per etype sample
    
    * Fix some bug and enable distributed training with per edge sample
    
    * fix
    
    * Now distributed training works
    
    * turn off some mxnet
    
    * turn off mxnet for some dist test
    
    * fix
    
    * upd
    
    * upd according to the comments
    
    * Fix
    
    * Fix test and now distributed works.
    
    * upd
    
    * upd
    
    * Fix
    
    * Fix bug
    
    * remove dead code.
    
    * upd
    
    * Fix
    
    * upd
    
    * Fix
    Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-71-112.ec2.internal>
    Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-2-66.ec2.internal>
    Co-authored-by: default avatarDa Zheng <zhengda1936@gmail.com>
    ba7e7cf9
rowwise_sampling.cc 10.9 KB