• Xin Yao's avatar
    [Sampling] Implement dgl.compact_graphs() for the GPU (#3423) · a8c81018
    Xin Yao authored
    * gpu compact graph template
    
    * cuda compact graph draft
    
    * fix typo
    
    * compact graphs
    
    * pass unit test but fail in training
    
    * example using EdgeDataLoader on the GPU
    
    * refactor cuda_compact_graph and cuda_to_block
    
    * update training scripts
    
    * fix linting
    
    * fix linting
    
    * fix exclude_edges for the GPU
    
    * add --data-cpu & fix copyright
    a8c81018
compact.h 1.68 KB