提交

2158 次代码提交

作者 SHA1 备注 提交日期
Quan Gan a55e499e1b update branch 0.8.0 2022-02-28 19:05:13 +00:00
Quan Gan c2b6376ed1 update version 2022-02-28 19:05:13 +00:00
Quan (Andy) Gan 3521fbe9ae [Build] Split spmm.cu and sddmm.cu for building on Windows (#3789)
* split files

* fix
2022-03-01 03:04:51 +08:00
Quan (Andy) Gan 6e1c69904e [Build or bug?] Fix VS2019 compilation error in randomwalk GPU kernel (#3788)
* Update randomwalk_gpu.cu

* Update randomwalk_gpu.cu
2022-02-28 22:02:23 +08:00
Xin Yao 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
Xin Yao 861666fa55 [Bugfix] Fix RGCN Example (#3785) 2022-02-28 18:55:06 +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 62e23bd5b1 [Bugs] Fix distributed example error and import error (#3783)
* fix

* raise an error

* fix docserver crash
2022-02-28 14:58:50 +08:00
Jinjing Zhou 7ab10348ed Sort recipe in DGL-Go (#3781)
* add

* remove

* fix

* rework the readme and some changes

* add png

* update png

* add recipe get

* sort recipe list and remove copy

* fix

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2022-02-28 13:27:02 +08:00
Minjie Wang 869991d7dc [Doc] Update README; add DGL-Go related 2022-02-28 11:44:55 +08:00
Minjie Wang e36cada2e8 [Doc] Update readme for DGL-Go 2022-02-28 10:53:16 +08:00
Jinjing Zhou 266b21e535 [DGL-Go] Change name to dglgo (#3778)
* add

* remove

* fix

* rework the readme and some changes

* add png

* update png

* add recipe get

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2022-02-28 10:40:01 +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 ccaa0bf282 [Doc] refine docstring and examples for reorder_graph (#3770)
* [Doc] refine docstring and examples for reorder_graph

* refine docstring
2022-02-24 17:31:00 +08:00
Rhett Ying 609a49a9ba [Doc] fix example in dataset adapter docstring (#3772) 2022-02-24 16:57:25 +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
sanchit-misra e7ad4c9c8b Fixes the bug when total_nnz is > integer limit (#3766) 2022-02-23 13:34:34 -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
Rhett Ying 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
nv-dlasalle bf649d940c [Bugfix][Performance] Update nccl to 2.11.4 (#3761)
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2022-02-23 13:51:41 +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
Quan (Andy) Gan 4f00d5ac6f [Bugfix] Fix graph being duplicated in multi-GPU and CPU dataloader workers (#3760)
* fix shared memory issue

* oops

* add explanation

* add explanation
2022-02-22 18:00:47 +08:00
Quan (Andy) Gan 3f138ebaaf [Bugfix] Bug fixes in new dataloader (#3727)
* fixes

* fix

* more fixes

* update

* oops

* lint?

* temporarily revert - will fix in another PR

* more fixes

* skipping mxnet test

* address comments

* fix DDP

* fix edge dataloader exclusion problems

* stupid bug

* fix

* use_uvm option

* fix

* fixes

* fixes

* fixes

* fixes

* add evaluation for cluster gcn and ddp

* stupid bug again

* fixes

* move sanity checks to only support DGLGraphs

* pytorch lightning compatibility fixes

* remove

* poke

* more fixes

* fix

* fix

* disable test

* docstrings

* why is it getting a memory leak?

* fix

* update

* updates and temporarily disable forkingpickler

* update

* fix?

* fix?

* oops

* oops

* fix

* lint

* huh

* uh

* update

* fix

* made it memory efficient

* refine exclude interface

* fix tutorial

* fix tutorial

* fix graph duplication in CPU dataloader workers

* lint

* lint

* Revert "lint"

This reverts commit 805484dd553695111b5fb37f2125214a6b7276e9.

* Revert "lint"

This reverts commit 0bce411b2b415c2ab770343949404498436dc8b2.

* Revert "fix graph duplication in CPU dataloader workers"

This reverts commit 9e3a8cf34c175d3093c773f6bb023b155f2bd27f.

Co-authored-by: xiny <xiny@nvidia.com>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2022-02-22 01:30:31 +08:00
ayasar70 7b9afbfa28 [Performance][GPU] Improving _SegmentMaskColKernel (#3745)
* 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

* timing repeatkernel

* clean

* clean

* clean

* updating _SegmentMaskColKernel

* Working on requests: removing sorted array check and adding comments to utility functions

* fixing lint issue

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>
2022-02-18 14:52:18 -08:00
Jinjing Zhou f247d29f6c Minor fix to DGL Enter (#3753)
* [Fix] Convert float64 to float32 when creating tensor

* fix

Co-authored-by: RhettYing <rhett_ying@qq.com>
Co-authored-by: Rhett Ying <85214957+Rhett-Ying@users.noreply.github.com>
2022-02-18 18:04:07 +08:00
Rhett Ying 5558ce2911 [Fix] Convert float64 to float32 when creating tensor (#3751)
* [Fix] Convert float64 to float32 when creating tensor

* refine docstring
2022-02-18 17:34:07 +08:00
Jinjing Zhou e424d29614 [CI] Use torch 1.9.0 in CI (#3752) 2022-02-18 16:35:24 +08:00
Jinjing Zhou 539335cece DGL Enter (#3690)
* add

* fix

* fix

* fix

* fix

* add

* add

* fix

* fix

* fix

* new loader

* fix

* fix

* fix for 3.6

* fix

* add

* add receipes and also some bug fixes

* fix

* fix

* fix

* fix receipies

* allow AsNodeDataset to work on ogb

* add ut

* many fixes for nodepred-ns pipeline

* receipe for nodepred-ns

* Update enter/README.md

Co-authored-by: Zihao Ye <zihaoye.cs@gmail.com>

* fix layers

* fix

* fix

* fix

* fix

* fix multiple issues

* fix for citation2

* fix comment

* fix

* fix

* clean up

* fix

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
Co-authored-by: Minjie Wang <minjie.wang@nyu.edu>
Co-authored-by: Zihao Ye <zihaoye.cs@gmail.com>
2022-02-18 15:55:41 +08:00
Rhett Ying 80fb4dbe26 [Doc] fix examples (#3747) 2022-02-17 18:46:13 +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
Mufei Li e9c3c0e8ad [Model] Simplify RGCN
* Update (#5)

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* FIx

* Try

* Update

* Update

* Update

* Fix

* Update

* Fix

* Fix

* Fix

* Fix

* Update

* Fix

* Update

* Update

* Update

* Fix

* Fix

* Update

* Update

* Update

* Update

* Fix

* Fix

* Fix

* Update

* Update

* Update

* Update

* Update

* Update README.md

* Update

* Fix

* Update

* Update

* Fix

* Fix

* Fix

* Update

* Update

* Update

Co-authored-by: Ubuntu <ubuntu@ip-172-31-6-240.us-west-2.compute.internal>

* Update

* Update

* Fix

* Update

* Update

* Update

* Fix

* Update

* Update

* Update

* Update

* Update

* Update

* CI

Co-authored-by: Ubuntu <ubuntu@ip-172-31-6-240.us-west-2.compute.internal>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-57-123.us-west-2.compute.internal>
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2022-02-17 15:33:27 +08:00
Rhett Ying 22272de652 [Fix] normalize by dst if edge_weight is None (#3744)
* [Fix] normalize by dst if edge_weight is None

* [Doc] fix math formula display issue
2022-02-17 13:51:37 +08:00
Minjie Wang 8a8d36e907 [Doc] add doc for DGLCSVDataset
[Doc] add doc for DGLCSVDataset
2022-02-17 10:59:53 +08:00
RhettYing 6ce3c178b0 fix typo 2022-02-17 02:11:16 +00:00
RhettYing 6106a99d0a refine 2022-02-17 02:00:10 +00:00
RhettYing 3b49370d3e refine doc 2022-02-17 00:52:20 +00:00
RhettYing 1722935223 add link for blitz 2022-02-17 00:52:20 +00:00
RhettYing 8d861d9443 refine especiaaly for data parser 2022-02-17 00:52:20 +00:00
RhettYing e0f054fb27 [Doc] add doc for DGLCSVDataset 2022-02-17 00:52:16 +00:00
Rhett Ying 39121dfdb8 [Feature] support non-numeric node_id/src_id/dst_id/graph_id and rena… (#3740)
* [Feature] support non-numeric node_id/src_id/dst_id/graph_id and rename CSVDataset

* change return value when iterate dataset

* refine data_parser

* force reload
2022-02-17 08:35:39 +08:00
Rhett Ying 42f8c8f335 [doc] fix module typo (#3734)
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2022-02-16 11:10:06 +08:00
Rhett Ying 97a6d740ad [doc] fix error in example results (#3735)
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2022-02-16 09:40:45 +08:00
Rhett Ying 4f7ad2582c [doc] fix error in example (#3736) 2022-02-15 18:44:15 +08:00
Mufei Li 8b8fd2c0be [Dataset] Add transform argument to built-in datasets (#3733)
* Update

* Fix

* Update
2022-02-15 16:45:47 +08: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 ab50eb9b4e Add unbuffered to distributed test (#3732)
Co-authored-by: Rhett Ying <85214957+Rhett-Ying@users.noreply.github.com>
2022-02-15 10:23:45 +08:00
Rhett Ying d8dd2bb542 [DOC] list transform in api reference (#3731)
* [DOC] list transform in api reference

* refine
2022-02-14 18:25:58 +08:00
Jonathan Broadbent 3b34a5a7ec Add links to RNAGlib in the DGL Readme (#3728)
* Add RNAGlib to examples and DGL-powered-projects

* Add RNAGlib to examples and DGL-powered-projects

* Add RNAGlib to examples and DGL-powered-projects
2022-02-11 15:00:18 -08:00