Skip to content

GitLab

Switch branch/tag
  1. 07 Dec, 2021 1 commit
  3. 06 Dec, 2021 4 commits
  5. 04 Dec, 2021 1 commit
  7. 03 Dec, 2021 2 commits
    • Israt Nisa's avatar
      [Feature] Add Min/max reducer in heterogeneous API for unary message functions (#3514) · cb0e1103
      Israt Nisa authored 
      

* min/max support for forward CPU heterograph

* Added etype with each argU values

* scatter_add needs fix

* added scatter_add_hetero. Grads dont match for max reducer

* storing ntype in argX

* fixing scatter_add_hetero

* hetero matches with torch's scatter add

* works copy_e forward+cpu

* added backward for copy_rhs

* Computes gradient for all node types in one kernel

* bug fix

* unnitest for max/min on CPU

* renamed scatter_add_hetero to update_grad_minmax_hetero

* lint check and comment out cuda call for max. Code is for CPU only

* lint check

* replace inf with zero

* minor

* lint check

* removed LIBXSMM code from hetro code

* fixing backward operator of UpdateGradMinMaxHetero

* removed backward from update_grad_minmax_hetero

* docstring

* improved docstring and coding style

* Added pass by pointer for output

* typos and pass by references

* Support for copy_rhs

* Added header <string>

* fix bug in copy_u_max

* Added comments and dimension check of all etypes

* skip mxnet check

* pass by pointer output arrays

* updated docstring
Co-authored-by: default avatarIsrat Nisa <nisisrat@amazon.com>
Co-authored-by: default avatarQuan (Andy) Gan <coin2028@hotmail.com>
      cb0e1103
    • Jinjing Zhou's avatar
      Bring back thrust for backward compatibility (#3562) · 769718df
      Jinjing Zhou authored
      769718df
  9. 02 Dec, 2021 3 commits
  11. 30 Nov, 2021 2 commits
  13. 29 Nov, 2021 3 commits
  15. 27 Nov, 2021 1 commit
  17. 24 Nov, 2021 1 commit
  19. 23 Nov, 2021 1 commit
  21. 22 Nov, 2021 2 commits
  23. 21 Nov, 2021 1 commit
  25. 19 Nov, 2021 1 commit
  27. 18 Nov, 2021 1 commit
  29. 17 Nov, 2021 3 commits
  31. 16 Nov, 2021 2 commits
  33. 15 Nov, 2021 2 commits
  35. 14 Nov, 2021 1 commit
    • Yang Su's avatar
      Fix User Guide EN and CN typo (#3495) · 863c2495
      Yang Su authored 
      

* Update graph-heterogeneous.rst

`tensor([0, 1, 2, 0, 1, 2])` should be output instead of code

* Update message-api.rst

`updata_all_example()` should be `update_all_example()`

* Update message-efficient.rst

`cat_feat` need to concatenate with `dim=1` for the # edge features to match # edges

* Update nn-construction.rst

all `max_pool` in the aggregator type of `SAGEConv` should be `pool` instead

* Update graph-heterogeneous.rst

`tensor([0, 1, 2, 0, 1, 2])` should be output instead of code

* Update message-api.rst

`updata_all_example()` should be `update_all_example()`

* Update message-efficient.rst

`cat_feat` need to concatenate with `dim=1` for the # edge features to match # edges

* Update nn-construction.rst

all `max_pool` in the aggregator type of `SAGEConv` should be `pool` instead

* Update nn-forward.rst

all `max_pool` in the aggregator type of `SAGEConv` should be `pool` instead

* Update nn-forward.rst

all `max_pool` in the aggregator type of `SAGEConv` should be `pool` instead
Co-authored-by: default avatarzhjwy9343 <6593865@qq.com>
      863c2495
  37. 12 Nov, 2021 1 commit
  39. 10 Nov, 2021 4 commits
  41. 09 Nov, 2021 2 commits
  43. 08 Nov, 2021 1 commit
    • yifeim's avatar
      [Model] Lda subgraph (#3206) · fe6e01ad
      yifeim authored 
      

* add word_ids and simplify

* simplify

* add word_ids to be removed later

* remove word_ids

* seems to work

* tweak

* transpose word_z

* add word_ids example

* check api compatibility

* improve compatibility

* update doc

* tweak verbose

* restore word_z layout; tweak

* tweak

* tweak doc

* word_cT

* use log_weight and some other tweaks

* rewrite README

* update equations

* rewrite for clarity and pass tests

* tweak

* bugfix import

* fix unit test

* fix mult to be the same as old versions

* tweak

* could be a bugfix

* 0/0=nan

* add doc_subgraph utility function

* minor cache optimization

* minor cache tweak

* add environmental variable to trade cache speed for memory

* update README

* tweak

* add sparse update pass unit test

* simplify sparse update

* improve low-memory efficiency

* tweak

* add sample expectation scores to allow resampling

* simplify

* update comment

* avoid edge cases

* bugfix pred scores

* simplify

* add save function
Co-authored-by: default avatarYifei Ma <yifeim@amazon.com>
Co-authored-by: default avatarQuan (Andy) Gan <coin2028@hotmail.com>
Co-authored-by: default avatarJinjing Zhou <VoVAllen@users.noreply.github.com>
      fe6e01ad