1. 12 Apr, 2022 2 commits
  2. 09 Apr, 2022 1 commit
  3. 05 Apr, 2022 1 commit
    • nv-dlasalle's avatar
      [Examples] Update graphsage multi-gpu example to use mutliple GPUs for... · 27a6eb56
      nv-dlasalle authored
      
      [Examples] Update graphsage multi-gpu example to use mutliple GPUs for validation and testing. (#3827)
      
      * Update graphsage multi-gpu example to use mutliple GPUs for validation and
      testing.
      
      * Remove argmax
      
      * Fix rebase error
      
      * Add more documentation to example and simplify
      
      * Switch to name shared memory
      
      * Add comment about how training is distributed
      
      * Restore iteration count
      
      * fix munmap error reporting for better error messages
      Co-authored-by: default avatarQuan (Andy) Gan <coin2028@hotmail.com>
      27a6eb56
  4. 30 Mar, 2022 2 commits
  5. 26 Mar, 2022 1 commit
  6. 25 Mar, 2022 1 commit
  7. 23 Mar, 2022 1 commit
  8. 11 Mar, 2022 1 commit
  9. 10 Mar, 2022 1 commit
  10. 07 Mar, 2022 1 commit
  11. 03 Mar, 2022 1 commit
  12. 01 Mar, 2022 1 commit
  13. 28 Feb, 2022 1 commit
  14. 27 Feb, 2022 1 commit
  15. 24 Feb, 2022 1 commit
  16. 23 Feb, 2022 1 commit
    • 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
  17. 22 Feb, 2022 1 commit
  18. 21 Feb, 2022 1 commit
    • Quan (Andy) Gan's avatar
      [Bugfix] Bug fixes in new dataloader (#3727) · 3f138eba
      Quan (Andy) Gan authored
      
      
      * fixes
      
      * fix
      
      * more fixes
      
      * update
      
      * oops
      
      * lint?
      
      * temporarily revert - will fix in another PR
      
      * more fixes
      
      * skipping mxnet test
      
      * address comments
      
      * fix DDP
      
      * fix edge dataloader exclusion problems
      
      * stupid bug
      
      * fix
      
      * use_uvm option
      
      * fix
      
      * fixes
      
      * fixes
      
      * fixes
      
      * fixes
      
      * add evaluation for cluster gcn and ddp
      
      * stupid bug again
      
      * fixes
      
      * move sanity checks to only support DGLGraphs
      
      * pytorch lightning compatibility fixes
      
      * remove
      
      * poke
      
      * more fixes
      
      * fix
      
      * fix
      
      * disable test
      
      * docstrings
      
      * why is it getting a memory leak?
      
      * fix
      
      * update
      
      * updates and temporarily disable forkingpickler
      
      * update
      
      * fix?
      
      * fix?
      
      * oops
      
      * oops
      
      * fix
      
      * lint
      
      * huh
      
      * uh
      
      * update
      
      * fix
      
      * made it memory efficient
      
      * refine exclude interface
      
      * fix tutorial
      
      * fix tutorial
      
      * fix graph duplication in CPU dataloader workers
      
      * lint
      
      * lint
      
      * Revert "lint"
      
      This reverts commit 805484dd553695111b5fb37f2125214a6b7276e9.
      
      * Revert "lint"
      
      This reverts commit 0bce411b2b415c2ab770343949404498436dc8b2.
      
      * Revert "fix graph duplication in CPU dataloader workers"
      
      This reverts commit 9e3a8cf34c175d3093c773f6bb023b155f2bd27f.
      Co-authored-by: default avatarxiny <xiny@nvidia.com>
      Co-authored-by: default avatarJinjing Zhou <VoVAllen@users.noreply.github.com>
      3f138eba
  19. 17 Feb, 2022 1 commit
  20. 09 Feb, 2022 1 commit
    • Xin Yao's avatar
      [Feature] CUDA UVA sampling for MultiLayerNeighborSampler (#3674) · 738e8318
      Xin Yao authored
      
      
      * implement pin_memory/unpin_memory/is_pinned for dgl.graph
      
      * update python docstring
      
      * update c++ docstring
      
      * add test
      
      * fix the broken UnifiedTensor
      
      * XPU_SWITCH for kDLCPUPinned
      
      * a rough version ready for testing
      
      * eliminate extra context parameter for pin/unpin
      
      * update train_sampling
      
      * fix linting
      
      * fix typo
      
      * multi-gpu uva sampling case
      
      * disable new format materialization for pinned graphs
      
      * update python doc for pin_memory_
      
      * fix unit test
      
      * UVA sampling for link prediction
      
      * dispatch most csr ops
      
      * update graphsage example to combine uva sampling and UnifiedTensor
      
      * update graphsage example to combine uva sampling and UnifiedTensor
      
      * update graphsage example to combine uva sampling and UnifiedTensor
      
      * update doc
      
      * update examples
      
      * change unitgraph and heterograph's PinMemory to in-place
      
      * update examples for multi-gpu uva sampling
      
      * update doc
      
      * fix linting
      
      * fix cpu build
      
      * fix is_pinned for DistGraph
      
      * fix is_pinned for DistGraph
      
      * update graphsage unsupervised example
      
      * update doc for gpu sampling
      
      * update some check for sampling device switching
      
      * fix linting
      
      * adapt for new dataloader
      
      * fix linting
      
      * fix
      
      * fix some name issue
      
      * adjust device check
      
      * add unit test for uva sampling & fix some zero_copy bug
      
      * fix linting
      
      * update num_threads in graphsage examples
      Co-authored-by: default avatarQuan (Andy) Gan <coin2028@hotmail.com>
      Co-authored-by: default avatarJinjing Zhou <VoVAllen@users.noreply.github.com>
      738e8318
  21. 07 Feb, 2022 1 commit
  22. 30 Jan, 2022 1 commit
    • Quan (Andy) Gan's avatar
      [Sampling] New sampling pipeline plus asynchronous prefetching (#3665) · 701b4fcc
      Quan (Andy) Gan authored
      * initial update
      
      * more
      
      * more
      
      * multi-gpu example
      
      * cluster gcn, finalize homogeneous
      
      * more explanation
      
      * fix
      
      * bunch of fixes
      
      * fix
      
      * RGAT example and more fixes
      
      * shadow-gnn sampler and some changes in unit test
      
      * fix
      
      * wth
      
      * more fixes
      
      * remove shadow+node/edge dataloader tests for possible ux changes
      
      * lints
      
      * add legacy dataloading import just in case
      
      * fix
      
      * update pylint for f-strings
      
      * fix
      
      * lint
      
      * lint
      
      * lint again
      
      * cherry-picking commit fa9f494
      
      * oops
      
      * fix
      
      * add sample_neighbors in dist_graph
      
      * fix
      
      * lint
      
      * fix
      
      * fix
      
      * fix
      
      * fix tutorial
      
      * fix
      
      * fix
      
      * fix
      
      * fix warning
      
      * remove debug
      
      * add get_foo_storage apis
      
      * lint
      701b4fcc
  23. 25 Jan, 2022 1 commit
  24. 23 Jan, 2022 1 commit
  25. 21 Jan, 2022 1 commit
  26. 20 Jan, 2022 1 commit
  27. 19 Jan, 2022 1 commit
  28. 15 Jan, 2022 1 commit
  29. 24 Dec, 2021 1 commit
  30. 20 Dec, 2021 2 commits
  31. 15 Dec, 2021 1 commit
    • Vasimuddin Md's avatar
      [DistGNN, Graph partitioning] Libra partition (#3376) · 78e0dae6
      Vasimuddin Md authored
      
      
      * added distgnn plus libra codebase
      
      * Dist application codes
      
      * added comments in partition code. changed the interface of partitioning call.
      
      * updated readme
      
      * create libra partitioning branch for the PR
      
      * removed disgnn files for first PR
      
      * updated kernel.cc
      
      * added libra_partition.cc and moved libra code from kernel.cc to libra_partition.cc
      
      * fixed lint error; merged libra2dgl.py and main_Libra.py to libra_partition.py; added graphsage/distgnn folder and partition script.
      
      * removed libra2dgl.py
      
      * fixed the lint error and cleaned the code.
      
      * revisions due to PR comments. added distgnn/tools contains partitions routines
      
      * update 2 PR revision I
      
      * fixed errors; also improved the runtime by 10x.
      
      * fixed minor lint error
      
      * fixed some more lints
      
      * PR revision II changed the interface of libra partition function
      
      * rewrite docstring
      Co-authored-by: default avatarQuan (Andy) Gan <coin2028@hotmail.com>
      78e0dae6
  32. 06 Dec, 2021 1 commit
  33. 30 Nov, 2021 1 commit
  34. 29 Nov, 2021 1 commit
  35. 24 Nov, 2021 1 commit
  36. 23 Nov, 2021 1 commit
  37. 19 Nov, 2021 1 commit