Skip to content

GitLab

Switch branch/tag
  1. 17 Feb, 2022 7 commits
  3. 16 Feb, 2022 2 commits
  5. 15 Feb, 2022 4 commits
    • Rhett Ying's avatar
      [doc] fix error in example (#3736) · 4f7ad258
      Rhett Ying authored
      4f7ad258
    • Mufei Li's avatar
      [Dataset] Add transform argument to built-in datasets (#3733) · 8b8fd2c0
      Mufei Li authored 
      * Update

* Fix

* Update
      8b8fd2c0
    • Israt Nisa's avatar
      [Feature] Gather mm (#3641) · b3d3a2c4
      Israt Nisa authored 
      

* init

* init

* working cublasGemm

* benchmark high-mem/low-mem, err gather_mm output

* cuda kernel for bmm like kernel

* removed cpu copy for E_per_Rel

* benchmark code from Minjie

* fixed cublas results in gathermm sorted

* use GPU shared mem in unsorted gather mm

* minor

* Added an optimal version of gather_mm_unsorted

* lint

* init gather_mm_scatter

* cublas transpose added

* fixed h_offset for multiple rel

* backward unittest

* cublas support to transpose W

* adding missed file

* forgot to add header file

* lint

* lint

* cleanup

* lint

* docstring

* lint

* added unittest

* lint

* lint

* unittest

* changed err type

* skip cpu test

* skip CPU code

* move in-len loop inside

* lint

* added check different dim length for B

* w_per_len is optional now

* moved gather_mm to pytorch/backend with backward support

* removed a_/b_trans support

* transpose op inside GEMM call

* removed out alloc from API, changed W 2D to 3D

* Added se_gather_mm, Separate API for sortedE

* Fixed gather_mm (unsorted) user interface

* unsorted gmm backward + separate CAPI for un/sorted A

* typecast to float to support atomicAdd

* lint typecast

* lint

* added gather_mm_scatter

* minor

* const

* design changes

* Added idx_a, idx_b support gmm_scatter

* dgl doc

* lint

* adding gather_mm in ops

* lint

* lint

* minor

* removed benchmark files

* minor

* empty commit
Co-authored-by: default avatarIsrat Nisa <nisisrat@amazon.com>
      b3d3a2c4
    • Jinjing Zhou's avatar
      Add unbuffered to distributed test (#3732) · ab50eb9b
      Jinjing Zhou authored 
      
Co-authored-by: default avatarRhett Ying <85214957+Rhett-Ying@users.noreply.github.com>
      ab50eb9b
  7. 14 Feb, 2022 1 commit
  9. 11 Feb, 2022 3 commits
  11. 10 Feb, 2022 1 commit
  13. 09 Feb, 2022 2 commits
    • Rhett Ying's avatar
      [Feature] Launch Long Live Servers and Multiple Client Groups (#3688) · fcd8ed9a
      Rhett Ying authored 
      * enable to launch multiple client groups sequentially

* launch simultaneously is enabled

* refine docstring

* revert unnecessary change

* [DOC] add doc for long live server

* refine

* refine doc

* refine doc
      fcd8ed9a
    • 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
  15. 07 Feb, 2022 4 commits
  17. 03 Feb, 2022 1 commit
  19. 30 Jan, 2022 3 commits
    • Rhett Ying's avatar
      [Fix] sleep for a while when launching clients which will connect to … (#3704) · 9c8c162a
      Rhett Ying authored 
      * [Fix] sleep for a while when launching clients which will connect to multiple servers

* pre-allocate more ports

* no multiple partitions on single machine
      9c8c162a
    • 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
    • Minjie Wang's avatar
      [Data] Utility function and class for converting a dataset for node prediction (#3695) · 5152a879
      Minjie Wang authored 
      * add ut

* add doc link

* install dep

* fix  ci

* fix ut; more comments

* remove deprecated attributes in rdf datasets; fix label feature name

* address comments

* fix ut for other frameworks
      5152a879
  21. 29 Jan, 2022 1 commit
  23. 28 Jan, 2022 5 commits
  25. 27 Jan, 2022 1 commit
  27. 26 Jan, 2022 3 commits
  29. 25 Jan, 2022 2 commits