* new functional for creating data splits in graph
* minor fix in data split implementation
* apply suggestions from code review
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
* refactoring + unit tests
* fix test file name
* move imports to the top
* Revert "fix test file name"
This reverts commit 126323e38c8de6fdc73c826dccc9048aed8f9634.
* remove nccl submodule
* address linter issues
---------
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
Co-authored-by: Hongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
* Bugfix for reverse edge issue.
* Resolved copy/paste transcription error
* lint
* Added unit tests for graph with not all reverses.
* linting + added backend device test
---------
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
Co-authored-by: Rhett Ying <85214957+Rhett-Ying@users.noreply.github.com>
* add CuGraphGATConv model
* lintrunner
* update model to reflect changes in make_mfg_csr(), move max_in_degree to forward()
* simplify pytest markers
* fall back to FG option for large fanout
* update error msg
* add feat_drop and activation options
* add residual option
* Update python/dgl/nn/pytorch/conv/cugraph_gatconv.py
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
* Update python/dgl/nn/pytorch/conv/cugraph_gatconv.py
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
* reset res_fc
---------
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
* subgraphx commit
* nits
* newline eof added
* lint fix
* test script updated to use default values
* lint fix
* graphs that are used for test cases are updated to a small graph
* lint formatted
* test paramter adj to complete the test under 20s
* lint fixes
---------
Co-authored-by: kxm180046 <kxm180046@utdallas.edu>
* update cugraph_relgraphconv
* update equality test
* update cugraph rgcn example
* update RelGraphConvAgg based on latest API changes
* enable fallback option to fg when fanout is large
---------
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
Some weird formatting occurred in the documentation for EGATConv, so this quick commit should fix that.
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
* enable sparse on windows and mac
* that was stupid
* let's see what's going on..
* [Sparse] Fix the import error on Mac OS.
When using template functions that are defined in source files from DGL,
the loader of MacOS somehow cannot find their definitions. This fix simply
avoids depending on template functions from DGL headers.
With this fix, the sparse tests all pass on the MAC environment.
* ok this is the problem
* make errors clearer
* uh
* test
* Update __init__.py
* disabling ddp on windows
---------
Co-authored-by: czkkkkkk <zekucai@gmail.com>
* Fix Column.record_stream(...) for unmaterialized tensors
* lint
* Record streams for indices on non-materialized columns
* add docstring
* fix for cpu index
* Record also stream on storage
* Always call self.storage.record_stream when storage is on GPU
* fix lint
---------
Co-authored-by: Xin Yao <xiny@nvidia.com>
* add SVD positional encoding
* modify importing module
* Fixed certain problems
* Change the test unit to a nonsigular one
* Fixed typo and make accord with lintrunner
* added svd_pe into dgl.rst
* Modified dgl.rst
* [Doc] Labor sampling improvement
When importance sampling option is used, the returned blocks include edge weights that need to be used during the message passing operation. So, I am including this in the documentation.
* Add documentation to weighted sampling case also
czkkkkkk