1. 21 Oct, 2021 1 commit
    • 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
  2. 10 Mar, 2020 1 commit
  3. 07 Mar, 2020 1 commit
  4. 28 Feb, 2020 1 commit
    • Quan (Andy) Gan's avatar
      [Feature] PinSAGE sampler (#1249) · c3a33407
      Quan (Andy) Gan authored
      
      
      * [WIP] PinSAGE operators
      
      * moved the edge remapping mess into C
      
      * some docstrings
      
      * lint
      
      * lint x2
      
      * lint x3
      
      * skip gpu test on topk
      
      * extend pinsage to any metapath
      
      * lint x4
      
      * addresses #1265
      
      * add always_preserve (fixes #1266) and fix a silly bug
      
      * disable gpu test on compaction
      
      * lint
      
      * fix a horrible bug and add more tests
      
      * lint
      
      * addresses comments
      
      * lint
      
      * bugfix
      
      * addresses comments
      Co-authored-by: default avatarMinjie Wang <minjie.wang@nyu.edu>
      c3a33407