• Minjie Wang's avatar
    [NN] Rework RelGraphConv and HGTConv (#3742) · 0227ddfb
    Minjie Wang authored
    * WIP: TypedLinear and new RelGraphConv
    
    * wip
    
    * further simplify RGCN
    
    * a bunch of tweak for performance; add basic cpu support
    
    * update on segmm
    
    * wip: segment.cu
    
    * new backward kernel works
    
    * fix a bunch of bugs in kernel; leave idx_a for future
    
    * add nn test for typed_linear
    
    * rgcn nn test
    
    * bugfix in corner case; update RGCN README
    
    * doc
    
    * fix cpp lint
    
    * fix lint
    
    * fix ut
    
    * wip: hgtconv; presorted flag for rgcn
    
    * hgt code and ut; WIP: some fix on reorder graph
    
    * better typed linear init
    
    * fix ut
    
    * fix lint; add docstring
    0227ddfb
entity_sample_multi_gpu.py 5.63 KB