Make the default value of parameter replace in docstring consistent with code
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
* 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: Israt Nisa <nisisrat@amazon.com>
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
* rgcn with new heterograph API
* added new apply_edge()
* optimized forward pass
* renaming from *hetero to *heteroAPI
Co-authored-by: Israt Nisa <nisisrat@amazon.com>
Co-authored-by: Da Zheng <zhengda1936@gmail.com>
* Based on issue #3436. Improving _SegmentCopyKernel s GPU utilization by switching to nonzero based thread assignment
* fixing lint issues
* Update cub for cuda 11.5 compatibility (#3468)
* fixing type mismatch
* tx guaranteed to be smaller than nnz. Hence removing last check
* minor: updating comment
* adding three unit tests for csr slice method to cover some corner cases
* working on repeat
* updating repeat kernel
* removing unnecessary parameter
* cleaning commented line
* cleaning time measures
* cleaning time measurement lines
Co-authored-by: Abdurrahman Yasar <ayasar@nvidia.com>
Co-authored-by: nv-dlasalle <63612878+nv-dlasalle@users.noreply.github.com>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
* Feat: support API "randomwalk_topk" in library
* Feat: use the new API "randomwalk_topk" for PinSAGESampler
* Minor
* Minor
* Refactor: modified codes as checker required
* Minor
* Minor
* Minor
* Minor
* Fix: checking errors in RandomWalkTopk
* Refactor: modified the docstring for randomwalk_topk
* change randomwalk_topk to internal
* fix
* rename
* Minor for pinsage.py
Co-authored-by: Quan Gan <coin2028@hotmail.com>
* Added SDDMMCOO_hetero support
* removed redundant CUDA kernels
* added benchmark for regression test
* fix
* fixed bug for single src node type
Co-authored-by: Israt Nisa <nisisrat@amazon.com>
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
* [Feature] enable async transfer in NodeDataLoader for homograph
* fix lint issues
* fix device choose when creating stream
* fix test on cpu only machine
* fix pin_memory config
* support homo only
* avoid creating stream in each step and sync via event
* fix lint
* enable graph copy on non-default stream
* fix lint
* refine arg description
* fix conflicts
* 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: zhjwy9343 <6593865@qq.com>
* squeeze node labels in FraudDataset
* fix RLModule
* update results in README.md
* fix KeyError in full graph training
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
* 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: Yifei Ma <yifeim@amazon.com>
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
* Based on issue #3436. Improving _SegmentCopyKernel s GPU utilization by switching to nonzero based thread assignment
* fixing lint issues
* Update cub for cuda 11.5 compatibility (#3468)
* fixing type mismatch
* tx guaranteed to be smaller than nnz. Hence removing last check
* minor: updating comment
* adding three unit tests for csr slice method to cover some corner cases
Co-authored-by: Abdurrahman Yasar <ayasar@nvidia.com>
Co-authored-by: nv-dlasalle <63612878+nv-dlasalle@users.noreply.github.com>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Jinjing Zhou