Israt Nisa
09ec5fd5e3
[Sparse] Improve API docs ( #4663 )
...
* replace with pow
* autosummary for attributes
* revert accidental push
* revert accidental push
* added example
Co-authored-by: Israt Nisa <nisisrat@amazon.com >
2022-09-30 12:34:42 +08:00
LuckyLiuM
eb729c5401
Add Deepwalk module ( #4562 )
...
* metapath2vec package
* fix bugs --metapath2vec package
* add unittest and fix bugs
* fix pyling messages
* del init.py
* fix bugs
* modify metapath2vec and add deepwalk
* deepwalk module
* del metapath2vec files
* Update
* Update
* Update
* Update
* Update
* Fix
Co-authored-by: Ubuntu <ubuntu@ip-172-31-9-26.ap-northeast-1.compute.internal >
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
2022-09-29 09:30:56 +08:00
Israt Nisa
fa05ccb961
Add sddmm operator in mock_sparse library ( #4579 )
...
* sddmm init
* SDDMM with N-D nonzero values
* drop support for vector shaped non zero elements
* address comments
* skip cpu test
* skip GPU test too
Co-authored-by: Israt Nisa <nisisrat@amazon.com >
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
2022-09-28 11:50:30 -04:00
Mufei Li
977b1ba448
[Sparse] Migration of Unary, Softmax, Matmul Ops ( #4616 )
...
* Update
* lint
* Fix
* Update
* Update
* Update
* update
* Fix
* Update
* Fix
* Fix
* Fix
* CI
* Update
* Update
* Update
* update test
Co-authored-by: Ubuntu <ubuntu@ip-172-31-9-26.ap-northeast-1.compute.internal >
2022-09-27 19:34:46 +08:00
czkkkkkk
544a2cd192
[Sparse] Add reduction operators ( #4600 )
...
* [Sparse] Add reduction operators
* Fix linting problem
* Fix linting problem
* Fix linting problem
* Fix docstring
* Fix test cases
* Fix linting problem
* Skip reduction unittests
2022-09-26 16:06:05 +08:00
czkkkkkk
db64cc3760
[Sparse] Add Transposition for SparseMatrix and DiagMatrix ( #4597 )
...
* [Sparse] Add Transposition
* Fix docstring
* Fix linting problem
* Minor fix
* Minor fix
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
2022-09-23 15:17:55 +08:00
Mufei Li
73b9f3a1da
[Sparse] Migrate DiagMatrix class ( #4567 )
...
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
Co-authored-by: Ubuntu <ubuntu@ip-172-31-9-26.ap-northeast-1.compute.internal >
2022-09-20 17:17:43 +08:00
Israt Nisa
cf03592743
Move mock version of dgl_sparse library to DGL main repo ( #4524 )
...
* init
* Add api doc for sparse library
* support op btwn matrices with differnt sparsity
* Fixed docstring
* addresses comments
* lint check
* change keyword format to fmt
Co-authored-by: Israt Nisa <nisisrat@amazon.com >
2022-09-14 12:08:20 -04:00
peizhou001
62af41c245
[Bug] Enable turn on/off libxsmm at runtime ( #4455 )
...
* enable turn on/off libxsmm at runtime by adding a global config and related API
Co-authored-by: Ubuntu <ubuntu@ip-172-31-19-194.ap-northeast-1.compute.internal >
2022-09-05 17:38:34 +08:00
Minjie Wang
3bcb268a36
[Doc] Update distributed chapter according to new pipeline ( #4275 )
...
* dist index chapter
* preproc chapter
* rst
* tools page
* partition chapter
* rst
* hetero chapter
* 7.1 step1
* add parmetis back
* changed based on feedback
* address comments
2022-08-17 13:30:04 +08:00
Rhett Ying
39987bc599
[Feature] enable graph partition book support canonical etypes ( #4343 )
...
* [Feature] enable graph partition book support canonical etypes
* fix lint
* fix lint
* add todo
* refine according to review comments
* fix lint
* refine naming
* revert PartitionPolicy __init__
* refine docstring
* fix doc string
2022-08-16 11:52:02 +08:00
Vibhu Jawa
9ae117d3cd
[Feature] Add to_cugraph and from_cugraph to DGL ( #4166 )
...
* Added to_cugraph and from_cugraph functionality
* fix from_cugraph example
* Addressed Reviews
* Fix linting
* Apply suggestions to docstrings from code review
Co-authored-by: Minjie Wang <minjie.wang@nyu.edu >
* Add API docs and remove `from_cugraph` alias
* move cugraph tests to tests/cugraph/test_basics.py
* remove ununsed imports from test_basics.py
* remove pytest.importorskip as no longer needed
Co-authored-by: Minjie Wang <minjie.wang@nyu.edu >
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
2022-07-07 11:52:17 +08:00
Rhett Ying
69226588a5
[Dist] defer to load node/edge feats ( #4143 )
...
* [Dist] defer to load node/edge feats
* fix lint
* Update python/dgl/distributed/partition.py
Co-authored-by: Minjie Wang <minjie.wang@nyu.edu >
* Update python/dgl/distributed/partition.py
Co-authored-by: Minjie Wang <minjie.wang@nyu.edu >
* fix lint
Co-authored-by: Minjie Wang <minjie.wang@nyu.edu >
2022-06-20 19:44:37 +08:00
ndickson-nvidia
5f04fc2b31
[Doc] Updated transform ops list in dgl.rst (Issue #4087 ) ( #4123 )
...
* * Added functions from dgl.transforms.functional that were missing from the list for documentation in dgl.rst
* * Sorted transform ops list in dgl.rst in alphabetical order
Co-authored-by: Xin Yao <xiny@nvidia.com >
2022-06-15 14:52:37 +08:00
RecLusIve-F
defa292bc0
[Dataset] Add Flickr and Yelp dataset ( #4099 )
...
* Add Flickr and Yelp dataset
* Update flickr.py
* update
* Update yelp.py
* Update yelp.py
* update
* Update yelp.py
* Update test_data.py
* Update yelp.py
* update
* Update test_data.py
* Update yelp.py
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
2022-06-14 16:14:12 +08:00
RecLusIve-F
89655cfda2
[Dataset] Add WikiCS Dataset ( #4035 )
...
* Fix bugs & Update dataset
* Update
* Update wikics.py
* Update wikics.py
* Update test_data.py
* Update wikics.py
* Update wikics.py
* Update wikics.py
* update
* Update module.py
* Update dgl.data.rst
* Update wikics.py
* Update wikics.py
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
2022-06-02 23:03:17 +08:00
Mufei Li
d9c25521bc
[Data] AsGraphPredDataset ( #4073 )
...
* Update
* CI
* Update
* Update
* Fix
* Fix
2022-06-02 16:18:48 +08:00
Mufei Li
bef993076e
[Bugfix] Fix PinSAGE Benchmark ( #4058 )
...
* Update
* Update
* Update dgl.data.rst
* CI
2022-05-28 00:31:57 +08:00
rudongyu
3fe5eea791
[NN] Label Propagation & Directional Graph Networks ( #4017 )
...
* add label propagation module
* fix prev bug in example
* add dgn
* fix linting and doc issues
* update label propagation & dgn
* update label propagation & dgn
* update example
* fix unit test
* fix agg heritage issue
* fix agg issue
* fix lint
* fix idx
* fix lp gpu issue
* Update
* Update
Co-authored-by: mufeili <mufeili1996@gmail.com >
2022-05-20 15:23:19 +08:00
Mufei Li
65b0b9e8c3
[Dataset & Transform] Synthetic Datasets for Explainability and SIGNDiffusion Transform ( #3982 )
...
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Fix
* Update
* Update
* Update
2022-05-16 22:32:09 +08:00
RecLusIve-F
b2ec5a2ea6
[Transform] Add FeatMask and RowFeatNormalizer transforms ( #3968 )
...
* [Model]P-GNN
* updata
* [Example]P-GNN
* Update README.md
* Add NodeFeatureMasking and NormalizeFeatures
* Update
* Update transforms.rst
* Update
* Update
* Update
* Update test_transform.py
* Update
* Update
* Update test_transform.py
* Update module.py
* Update module.py
* Update module.py
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
2022-05-07 11:59:39 +08:00
Mufei Li
248bece633
[NN] GINEConv ( #3934 )
...
* Update
* Update
* Update
* Update
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
2022-04-25 14:41:05 +08:00
Quan (Andy) Gan
4375c2d7e4
[Doc] Fix documentation in dgl.multiprocessing namespace ( #3929 )
...
* fix docs
* remove
* oh
* fix
2022-04-16 00:35:06 +08:00
Quan (Andy) Gan
e06e63d5d5
[Feature] Enable UVA sampling with CPU indices ( #3892 )
...
* enable UVA sampling with CPU indices
* add docs
* add more docs
* lint
* fix
* fix
* better error message
* use mp.Barrier instead of queues
* revert
* revert
* oops
* revert dgl.multiprocessing.spawn
* Update pytorch.py
2022-04-12 20:56:51 +08:00
rudongyu
dc5035b1f3
[NN] Add EGNN & PNA ( #3901 )
...
* add EGNN & PNA
* fix egnn issues
* fix pna issues
* update pna conv
* add doc strings
* update pnaconv
* fix unused args issue
* fix moment aggregation issue
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
2022-04-07 17:44:57 +08:00
Krzysztof Sadowski
4c7bd314b4
[Feature] Radius Graph ( #3829 )
...
* radius graph
* remove trailing whitespaces from docs
* disable invalid name for transform func
* disable radius graph module invalid name
* move pylint disable before init
* fix missing nodes from point set
* update docs indexing
* add compute mode as optional param
* radius graph test
* remove trailing whitespaces
* fix precision when comparing tensors
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
2022-03-30 17:23:29 +08:00
rudongyu
290b7c25c2
[Transform] Random Walk Positional Encoding & Laplacian Positional Encoding ( #3869 )
...
* [Transform] Random Walk Positional Encoding & Laplacian Positional Encoding
* Update python/dgl/transforms/module.py
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
* Update python/dgl/transforms/functional.py
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
* Update python/dgl/transforms/functional.py
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
* Update python/dgl/transforms/functional.py
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
* Update python/dgl/transforms/functional.py
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
* Update python/dgl/transforms/functional.py
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
* Update python/dgl/transforms/functional.py & python/dgl/transforms/module.py
* update python/dgl/transforms/functional.py
* update doc indexing
* update unit test
* support weighted graphs for random_walk_pe
* change reference
* fix linter errors
* fix datatype compatibility with MXNet & TF
* fix device errors
* fix result precision errors
* change tensorflow abs api
* reboost CI
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
2022-03-30 15:59:28 +08:00
Mufei Li
9fee20b91d
[Sampler] [Example] SAINTSampler and Simplify GraphSAINT Example ( #3879 )
...
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Fix
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com >
2022-03-30 13:51:37 +08:00
Minjie Wang
ae3316c846
[Doc] Update doc of dataloading package. ( #3886 )
...
* wip: dataloading doc
* update dataloading package doc and many others
* lint
2022-03-26 21:59:02 +08:00
Rhett Ying
16409ff852
[Doc] fix display issue of node/edge_split ( #3858 )
...
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com >
2022-03-25 09:18:08 +08:00
Mufei Li
8005978e1e
[NN] Grouped reversible residual connections for GNNs ( #3842 )
...
* Update
* Fix
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
2022-03-23 16:58:39 +08:00
Mufei Li
ec64581c96
[Doc] Misc fix for doc ( #3782 )
...
* Update nn-pytorch.rst
* Update dgl.data.rst
* Update dgl.rst
2022-02-28 16:03:32 +08:00
Quan (Andy) Gan
d41d07d0f6
[Doc and bugfix] Add docs and user guide and update tutorial for sampling pipeline ( #3774 )
...
* huuuuge update
* remove
* lint
* lint
* fix
* what happened to nccl
* update multi-gpu unsupervised graphsage example
* replace most of the dgl.mp.process with torch.mp.spawn
* update if condition for use_uva case
* update user guide
* address comments
* incorporating suggestions from @jermainewang
* oops
* fix tutorial to pass CI
* oops
* fix again
Co-authored-by: Xin Yao <xiny@nvidia.com >
2022-02-28 02:21:51 +08:00
Mufei Li
3bd5a9b6d1
[Doc] Fix Doc ( #3777 )
...
* Update
* Update
* Update
* Update
2022-02-25 17:21:06 +08:00
Mufei Li
6d9433b04f
[Transform] [Doc] Rename transform to transforms and update doc ( #3765 )
...
* Update
* Update
* Update
* Fix
* Update
* Update
* Update
* Fix
2022-02-24 21:28:15 +08:00
Rhett Ying
2f4146a4c2
[Dataset] add train_idx/val_idx/test_idx as dataset attributes ( #3769 )
...
* add train_idx/val_idx/test_idx as dataset attributes
* refine docstring
2022-02-24 15:17:40 +08:00
Rhett Ying
4b182b9063
toctree typo for dataset adapters ( #3764 )
...
* toctree typo for dataset adapters
* Update docs/source/api/python/dgl.data.rst
2022-02-23 18:46:19 +08:00
Minjie Wang
0227ddfb66
[NN] Rework RelGraphConv and HGTConv ( #3742 )
...
* 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
2022-02-23 12:10:20 +08:00
dddg617
9e358dfe66
[NN] HeteroLinear and HeteroEmbedding ( #3678 )
...
* modify hetero
* modify rst document
* update hetero
* update hetero
* update hetero
* update hetero
* Update
* Update
* Update
* Update
* 20220216
* Update
* Update
* Fix
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
Co-authored-by: ShelkerX <925089962@qq.com >
2022-02-17 16:16:51 +08:00
RhettYing
6106a99d0a
refine
2022-02-17 02:00:10 +00:00
RhettYing
e0f054fb27
[Doc] add doc for DGLCSVDataset
2022-02-17 00:52:16 +00:00
Israt Nisa
b3d3a2c4b0
[Feature] Gather mm ( #3641 )
...
* 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: Israt Nisa <nisisrat@amazon.com >
2022-02-14 22:27:23 -05:00
Jinjing Zhou
dc78e11c3f
[Dataset] Adapter to convert a dataset for link prediction task ( #3699 )
...
* add ut
* add doc link
* install dep
* add
* Revert "install dep"
This reverts commit e574a83771wmjlyjemaine@gmail.com >
2022-02-11 16:59:14 +08:00
Quan (Andy) Gan
701b4fccc2
[Sampling] New sampling pipeline plus asynchronous prefetching ( #3665 )
...
* 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
2022-01-30 16:13:00 +08:00
Minjie Wang
5152a87982
[Data] Utility function and class for converting a dataset for node prediction ( #3695 )
...
* 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
2022-01-30 14:31:37 +08:00
Mufei Li
2b98e764e8
[Transform] Modules for Augmentation ( #3668 )
...
* Update
* Update
* Fix
* Update
* Update
* Update
* Update
* Fix
* Update
* Update
* Update
* Update
* Fix lint
* lint
* Update
* Update
* lint fix
* Fix CI
* Fix
* Fix CI
* Update
* Fix
* Update
* Update
* Augmentation (#10 )
* Update
* PPR
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* CI
* lint
* lint
* Update
* Update
* Fix AddEdge
* try import
* Update
* Fix
* CI
Co-authored-by: Ubuntu <ubuntu@ip-172-31-31-136.us-west-2.compute.internal >
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
2022-01-25 20:29:46 +08:00
Jeremy Goh
8f5baa15e4
[Doc] Fix spelling, references and update info on building docs ( #3682 )
...
* Fix ref to message-passing guide
* Fix pygments and spacing
* Update build documentation steps in README.md
* Use links
* Adjust parameters in SAGEConv docstring in same order as init
* Fix spelling error
* Change doc link
2022-01-25 17:22:18 +08:00
Xin Yao
40b44a43a9
[Feature] Pin dgl.graph to the page-locked memory ( #3616 )
...
* implement pin_memory/unpin_memory/is_pinned for dgl.graph
* update python docstring
* update c++ docstring
* add test
* fix the broken UnifiedTensor
* eliminate extra context parameter for pin/unpin
* fix linting
* fix typo
* disable new format materialization for pinned graphs
* update python doc for pin_memory_
* fix unit test
* update doc
* change unitgraph and heterograph's PinMemory to in-place
* update comments for NDArray's PinMemory_ and PinData
* update doc
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com >
2022-01-21 16:13:12 +08:00
Mufei Li
48cbea720e
[Transform] Module Interface for Transform ( #3636 )
...
* Update
* Update
* Fix
* Update
* Update
* Update
* Update
* Fix
* Update
* Update
* Update
* Update
* Fix lint
* lint
* Update
* Update
* lint fix
* Fix CI
* Fix
* Fix CI
* Update
* Fix
* Update
* Update
* resolve conflict
* Fix CI
Co-authored-by: Ubuntu <ubuntu@ip-172-31-31-136.us-west-2.compute.internal >
2022-01-17 17:45:07 +08:00
Quan (Andy) Gan
90f10b31cb
[Feature] Negative sampling ( #3599 )
...
* first commit
* a bunch of fixes
* add unique
* lint
* lint
* lint
* address comments
* Update negative_sampler.py
* fix
* description
* address comments and fix
* fix
* replace unique with replace
* test pylint
* Update negative_sampler.py
2022-01-07 18:26:14 +08:00
