Skip to content

GitLab

Switch branch/tag
  1. 09 Aug, 2022 1 commit
  3. 01 Aug, 2022 1 commit
  5. 29 Jul, 2022 1 commit
    • Xin Yao's avatar
      [Feature] Add CUDA Weighted Neighborhood Sampling (#4064) · 86c81b4e
      Xin Yao authored 
      

* add weighted sampling without replacement (A-Chao)

* improve Algorithm A-Chao with block-wise prefix sum

* correctly fill out_idxs

* implement weighted sampling with replacement

* small fix

* merge host-side code of weighted/uniform sampling

* enable unit tests for cuda weighted sampling

* move thrust/cub wrapper to the cmake file

* update docs accordingly

* fix linting

* fix linting

* fix unit test

* Bump external CUB/Thrust versions

* Fix code style and update description of algorithm design

* [Feature] GPU support weighted graph neighbor sampling
commit by pengqirong(OPPO)

* merge pengqirong's implementation

* revert the change to cub and thrust

* fix linting

* use DeviceSegmentedSort for better performance

* add more comments

* add necessary notes

* add necessary notes

* resolve some comments

* define THRUST_CUB_WRAPPED_NAMESPACE

* fix doc
Co-authored-by: default avatar彭齐荣 <657017034@qq.com>
      86c81b4e
  7. 27 Jul, 2022 1 commit
  9. 26 Jul, 2022 1 commit
  11. 14 Jul, 2022 1 commit
  13. 01 Jul, 2022 2 commits
  15. 30 Jun, 2022 2 commits
  17. 29 Jun, 2022 2 commits
  19. 28 Jun, 2022 1 commit
  21. 23 Jun, 2022 1 commit
  23. 14 Jun, 2022 1 commit
  25. 12 Jun, 2022 1 commit
  27. 07 Jun, 2022 1 commit
  29. 02 Jun, 2022 2 commits
  31. 28 May, 2022 1 commit
  33. 25 May, 2022 1 commit
  35. 24 May, 2022 1 commit
  37. 23 May, 2022 1 commit
  39. 17 May, 2022 1 commit
    • ndickson-nvidia's avatar
      [Feature] Added floating-point conversion functions to dgl.transforms.functional (#3890) · e0e8736f
      ndickson-nvidia authored 
      

* * Added half_(), float_(), and double_() functions to DGLHeteroGraph, HeteroNodeDataView, and HeteroEdgeDataView, for converting floating-point tensor data to float16, float32, or float64 precision

* * Extracted out private functions for floating-point type conversion, to reduce code duplication

* * Added test for floating-point data conversion functions, half_(), float_(), and double_()

* * Moved half_(), float_(), and double_() functions from HeteroNodeDataView and HeteroEdgeDataView to Frame class

* * Updated test_float_cast() to use dgl.heterograph instead of dgl.graph

* Added to CONTRIBUTORS.md

* * Changed data type conversion to be deferred until the data is accessed, to avoid redundant conversions of data that isn't used.

* * Addressed issues flagged by linter

* * Worked around a bug in the old version of mxnet that's currently used for DGL testing

* * Only defer Column data type conversion if there is a pending device transfer or index sampling to be done.  This is expected to be the desired behaviour based on discussions of a few use cases, as described in the comments.

* * Moved floating-point feature data conversion functions to dgl.transforms.functional
* Changed them from in-place behaviour to shallow copy (out-of-place) behaviour

* * Fixed linter issues

* * Removed lines that unintentionally added to_half, to_float, and to_double to DGLHeteroGraph
* Moved _init_api line to the end of the file again

* * Removed one of the two leading underscores from Frame.__astype_float, making it not fully private
Co-authored-by: default avatarnv-dlasalle <63612878+nv-dlasalle@users.noreply.github.com>
      e0e8736f
  41. 16 May, 2022 3 commits
  43. 07 May, 2022 1 commit
  45. 23 Apr, 2022 1 commit
  47. 12 Apr, 2022 2 commits
  49. 11 Apr, 2022 1 commit
  51. 09 Apr, 2022 1 commit
  53. 31 Mar, 2022 1 commit
  55. 30 Mar, 2022 1 commit
  57. 24 Mar, 2022 1 commit
  59. 24 Feb, 2022 2 commits
  61. 23 Feb, 2022 2 commits
    • Rhett Ying's avatar
      [Dataset][Fix] Allow non-numeric values and some fix on doc (#3757) · 8db2dd33
      Rhett Ying authored 
      * [Fix] be able to parse ids if numeric and non-numeric values are used together

* add required package info and cache note into docstring

* duplicate node id is not allowed
      8db2dd33
    • 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