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
bd311aa68d
[doc] fix invalid link in user guide ( #4468 )
2022-08-25 23:16:44 +08:00
7a41c126f6
[Doc] Change random.py to random_partition.py in guide on distributed partition pipeline ( #4438 )
...
* Update distributed-preprocessing.rst
* Update
Co-authored-by: Ubuntu <ubuntu@ip-172-31-9-26.ap-northeast-1.compute.internal >
2022-08-22 18:12:55 +08:00
bc2fef9c32
[Doc] Misc Fix for User Guide 7.1 Data Preprocessing ( #4433 )
...
* Update
* rollback for partition_algo/random.py
Co-authored-by: Ubuntu <ubuntu@ip-172-31-20-21.us-west-2.compute.internal >
2022-08-18 20:27:40 +08:00
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
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
86c81b4e92
[Feature] Add CUDA Weighted Neighborhood Sampling ( #4064 )
...
* 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: 彭齐荣 <657017034@qq.com >
2022-07-29 11:08:48 +08:00
7e6a6b4ac0
[Feature] Add CUDA Weighted Randomwalk Sampling ( #4243 )
...
* [Feature] Add CUDA Weighted Randomwalk Sampling
* [Feature] Add CUDA Weighted Randomwalk Sampling
* [Feature] Add CUDA Weighted Randomwalk Sampling
* [Feature] Add CUDA Weighted Randomwalk Sampling
* fix empty prob array && enable non-uniform for restart && enable unit tests
* update doc and guide for randomwalk and pinsage
* update comments
Co-authored-by: zhenliangqiu <ubuntu@ip-172-31-24-245.ap-southeast-1.compute.internal >
Co-authored-by: xiny <xiny@nvidia.com >
2022-07-26 13:42:39 +08:00
05d9d4969b
Update ( #4277 )
...
Co-authored-by: Ubuntu <ubuntu@ip-172-31-53-142.us-west-2.compute.internal >
2022-07-21 13:27:42 +08:00
5c76e47f78
[Docs] fix ffi docs ( #4248 )
...
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
2022-07-14 13:20:42 +08:00
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
32f12ee19e
[Doc] Unify the minimal versions required for PyTorch/TensorFlow/MXNet ( #4180 )
2022-06-29 18:37:19 +08:00
d1f6f3a848
[Doc] fix a bug in guide_cn ( #4149 )
...
Co-authored-by: Xin Yao <xiny@nvidia.com >
2022-06-24 13:08:07 +08:00
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
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
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
92063d88f0
fix indent in minibatch-custom-sampler.rst ( #4063 )
...
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
2022-06-03 01:00:31 +08:00
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
d9c25521bc
[Data] AsGraphPredDataset ( #4073 )
...
* Update
* CI
* Update
* Update
* Fix
* Fix
2022-06-02 16:18:48 +08:00
bef993076e
[Bugfix] Fix PinSAGE Benchmark ( #4058 )
...
* Update
* Update
* Update dgl.data.rst
* CI
2022-05-28 00:31:57 +08:00
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
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
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
248bece633
[NN] GINEConv ( #3934 )
...
* Update
* Update
* Update
* Update
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
2022-04-25 14:41:05 +08:00
4375c2d7e4
[Doc] Fix documentation in dgl.multiprocessing namespace ( #3929 )
...
* fix docs
* remove
* oh
* fix
2022-04-16 00:35:06 +08:00
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
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
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
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
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
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
e963256882
[Doc] Update message-api.rst ( #3844 )
...
fix #3812
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com >
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
2022-03-25 23:22:20 +08:00
7d4160869c
[Bug] Fix multiple issues in distributed multi-GPU GraphSAGE example ( #3870 )
...
* fix distributed multi-GPU example device
* try Join
* update version requirement in README
* use model.join
* fix docs
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com >
2022-03-25 22:04:02 +08:00
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
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
8425c936c2
update doc for gpu&uva sampling ( #3787 )
...
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com >
2022-02-28 21:17:32 +08:00
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
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
3bd5a9b6d1
[Doc] Fix Doc ( #3777 )
...
* Update
* Update
* Update
* Update
2022-02-25 17:21:06 +08:00
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
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
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
8db2dd3355
[Dataset][Fix] Allow non-numeric values and some fix on doc ( #3757 )
...
* [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
2022-02-23 17:26:12 +08:00
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
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
6106a99d0a
refine
2022-02-17 02:00:10 +00:00
3b49370d3e
refine doc
2022-02-17 00:52:20 +00:00
8d861d9443
refine especiaaly for data parser
2022-02-17 00:52:20 +00:00
e0f054fb27
[Doc] add doc for DGLCSVDataset
2022-02-17 00:52:16 +00:00
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
peizhou001