文件历史

2100 次代码提交

作者 SHA1 备注 提交日期
xiang song(charlie.song) 4464b9734c Fix kg dataset len() (#2475)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-59-204.ec2.internal>
2020-12-31 23:33:21 +08:00
Da Zheng 3d1f2e870a delete shared memory when receiving signals. (#2419)
* delete shared memory when receive signal.

* rename.

* fix lint.

* fix lint.

* fix compile.

* Fix.

* we need to report error if the shared memory exist.

* disable tensorflow test for shared memory.

* revert.

Co-authored-by: Ubuntu <ubuntu@ip-172-31-2-202.us-west-1.compute.internal>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2020-12-26 10:41:10 -08:00
Quan (Andy) Gan e805470101 addressing post-merge comments (#2455) 2020-12-26 17:58:15 +08:00
Quan (Andy) Gan 9a7235faf2 [Performance] Use allocator from PyTorch if possible (#2328)
* first commit

* some thoughts

* move around

* more commit

* more fixes

* now it uses torch allocator

* fix symbol export error

* fix

* fixes

* test fix

* add script

* building separate library per version

* fix for vs2019

* more fixes

* fix on windows build

* update jenkinsfile

* auto copy built dlls for windows

* lint and installation guide update

* fix

* specify conda environment

* set environment for ci

* fix

* fix

* fix

* fix again

* revert

* fix cmake

* fix

* switch to using python interpreter path

* remove scripts

* debug

* oops sorry

* Update index.rst

* Update index.rst

* copies automatically, no need for this

* do not print message if library not found

* tiny fixes

* debug on nightly

* replace add_compile_definitions to make CMake 3.5 happy

* fix linking to wrong lib for multiple pytorch envs

* changed building strategy

* fix nightly

* fix windows

* fix windows again

* setup bugfix

* address comments

* change README
2020-12-25 13:57:51 +08:00
Zihao Ye baace710cf [documentation] Improve the docstring of global pooling layer (#2437)
* upd

* fix-link

* lint
2020-12-21 10:12:54 +08:00
Mufei Li 492ad9be81 [Fix] Avoid Overloading Nonzero for PyTorch Backend (#2434)
* Update gatedgraphconv.py

* Update entity_classify.py

* Update data-process.rst

* Update reading_data.py

* Update data-process.rst

* Update utils.py

* Update knowledge_graph.py

* Update entity_classify.py

* Update rdf.py

* Update entity_classify_mb.py

* Update test_classify.py

* Update tensor.py

* Update sparse.py

* Update entity_classify_mp.py

* Update 6_line_graph.py
2020-12-20 13:12:17 +08:00
xiang song(charlie.song) 9d7bf4ea89 Fix citation data loading (#2421)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-57-25.ec2.internal>
2020-12-14 14:58:01 +08:00
张天启 1c976222d5 fix doc mistake in dgl/core.py (#2404)
The variable name 'eid' should be 'nid'
2020-12-08 22:05:55 +08:00
Mufei Li 013d145663 [NN] Attention Retrieval for NN Modules (#2397)
* Update

* Update
2020-12-07 23:42:25 +08:00
Xiangkun Hu fcf1193579 [Doc] Hidden implementations that may cause confusion to the user. (#2378)
* PPIDataset

* Revert "PPIDataset"

This reverts commit 264bd0c960cfa698a7bb946dad132bf52c2d0c8a.

* data pipeline user guide

* remove chapter numbers

* Update data.rst

* image in dataset userguide

* update links using ref

* modify the link of save_graphs and load_graphs in dataset user guide

* move image to s3 server.

* fix links and ref

* Hide implementations that may cause confusions to the user
2020-12-03 13:09:19 +08:00
Zihao Ye 98e11533e2 [hotfix] Produce empty tensor in data frame for emtpy graphs. (#2387)
* upd

* lint

* lint1
2020-12-03 12:00:29 +08:00
Quan (Andy) Gan 38077e298b installation and doc fix (#2379) 2020-11-30 19:47:40 +08:00
Mufei Li edb978772c [Doc, UG, Tutorial] Misc Fix (#2376)
* Update

* Update

* Update

* Update

* Update
2020-11-30 13:40:10 +08:00
Zihao Ye 6b02babbad [doc] Add docstring for segment reduce. (#2375) 2020-11-27 18:04:13 +08:00
Quan (Andy) Gan 00edb94982 [Performance] Accelerate batching (#2363)
* speed up batching

* more fix

* lint

* fix
2020-11-24 17:39:00 +08:00
Zihao Ye 58775ada1f Remove redundant print information in #2361 (#2362) 2020-11-23 03:41:58 +08:00
Zihao Ye 3adbfa1812 [Performance] Use segment operators for graph readout. (#2361)
* upd

* upd

* update

* upd

* upd

* upd

* fix

* lint

* lint

* pylint

* doc
2020-11-23 01:45:33 +08:00
Mufei Li 45e3e9af2c [WIP] Misc Fix (#2359)
* Update

* Update

* Update
2020-11-19 23:54:55 +08:00
vill fe4ada23f9 Update knowledge_graph.py (#2329)
The l in "bool" is lost.

Co-authored-by: Zihao Ye <expye@outlook.com>
2020-11-10 17:33:55 +08:00
Mufei Li 812f597096 Update appnpconv.py (#2324) 2020-11-10 15:37:30 +08:00
Minjie Wang 4fb0241bfb [CUDA] Add CUDA11 support (#2308)
* add support for cuda 11

* fix inc bug in pytorch 1.8

* poke ci

* fix

* small fix

* try fix

* try fix

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2020-11-07 22:16:51 +08:00
nv-dlasalle 6eda605ccd [BugFix][Dataloading] Disable AsyncTransferer copying to anywhere but the GPU (#2323)
* Disable copying for anywhere but the GPU

* Remove unused import and remove references to transferring from the GPU from the docs

* Skip gpu test in cpu mode

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2020-11-07 21:34:54 +08:00
Xiangkun Hu 5f19876358 [Bugfix] GINDataset check whether labels are all valid (#2319)
* PPIDataset

* Revert "PPIDataset"

This reverts commit 264bd0c960cfa698a7bb946dad132bf52c2d0c8a.

* check whether the original labels are valid

* add unit test for gin dataset

* bug of asscalar in mxnet<=1.6

* mxnet<=1.6 asscalar requires ndarray to be shaped (1,)

* skip gpu while testing datasets

* Update test_data.py

* test of gin dataset takes too much time
2020-11-05 09:35:10 +08:00
nv-dlasalle c6890c231a [Performance] In HeteroNodeView, build arange on target device, instead of on CPU and copying it (#2266)
* Build arange on target device

* Utilize arange device in viewpy:HeteroNodeView.__call__

* Work around uint64 error in TF to_dlpack

* Restore else clause

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Da Zheng <zhengda1936@gmail.com>
2020-11-02 14:03:26 -08:00
nv-dlasalle d453d72dd1 [Doc][Dataloading] Expand documentation of AsyncTransferer (#2313)
* Update docs

* Make non-default streams non-blocking
2020-11-02 12:28:30 -08:00
nv-dlasalle f673fc2553 [Dataloading] Add class for copying tensors to/from the GPU on a non-default stream (#2284)
* Add async transferer class

* Add async ndarray copy interface

* Add python bindings

* Fix comment

* Add python class

* Fix linting issues

* Add python unit test

* Update python interface

* move async_transferer to cuda only directory

* Fix linting issue

* Move out of contrib

* Add doc strings

* Move test compute from backend

* Update comment

* Fix test naming

* Fix argument usage

* Wrap/unwrap backend parameters

* Move to dataloading

* Move to 'dataloading'

* Make GPU/CPU compatible

* Fix unit tests

* Add docs

* Use only backend interface for datamovement in unit test
2020-10-30 07:29:06 -07:00
chwan-rice 269983dbcd [Fix] typo (#2265)
* upd

* upd

* upd

* upd

* upd

* fix OpenMP compatibility issues

* typo

* partition

* misc

* fix typo

* num_parts=1

* import torch

* long

* print info

* print info

* print info

* upd

* remove debug code

* revert partition.py

* fix cut count

* fix cut count

* Revert "fix cut count"

This reverts commit 10926b4fd48f45c8f1ddb58be7db6c22e653effd.

* Revert "fix cut count"

This reverts commit 76465283bef093a2b4209ad70dd15d2437b2ec8a.

* type of deprecate

* typo in deprecate info

* fix typo

* use cv for partitioning

* CE

* no message

* revert

* typo

* no message

* no message

* no message

* practice

* dmlc-core

* DGLGraph

Co-authored-by: yzh119 <expye@outlook.com>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2020-10-29 23:05:36 +08:00
ZenoTan da9f14dff3 Fix typos (#2285) 2020-10-19 12:24:34 -07:00
nv-dlasalle 8a4520d55a [Bugfix][Performance] Fix Column to copy 'self.index' to location of 'self.storage' only on read, and use kwargs. (#2267)
* Pass framework arguments to copy of frames

* Copy for subframe

* Fix mismatched contexts

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2020-10-09 09:58:31 -07:00
Sami Issaadi 8bbc84e04f [bugfix] Fix crash due to unfound attribute (#2262)
An unexpected `AttributeError` exception is launched whenever the user defines `save_dir` on `DGLDataset` instantiation and uses `self._save_dir`  within `has_cache` method.
This commit fixes the issue.
2020-10-09 09:44:38 +08:00
Zihao Ye c341520d5b [bugfix] Fix the behavior of min/max reducer for 1-dim dta. (#2250)
* udp

* add test

* udp

* fix mxnet
2020-09-29 17:27:16 +08:00
Junki Ishikawa 6b0d42db15 Fix typo (#2235)
Co-authored-by: Chao Ma <mctt90@gmail.com>
2020-09-25 16:40:22 +08:00
Mingjian Wen 25a8e3f181 Fix segment_reduce() ignoring tailing 0 segments (#2228)
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2020-09-24 15:07:45 +08:00
Mufei Li 40caf1ab50 [HeteroGraph] Fix the failure of apply_nodes when the input function changes feature size for all nodes (#2223)
* Fix

* Fix
2020-09-24 08:54:28 +08:00
Quan (Andy) Gan 49e96970e2 [Doc] fix typos in PyTorch DataLoaders (#2216)
* fix doc

* fix doc
2020-09-20 13:21:05 +08:00
Mufei Li 0b00562d1f Update gindt.py (#2212) 2020-09-18 19:49:50 +08:00
Mufei Li f3f646bbfd [Doc and User Guide] Minor Fix (#2205)
* Fix doc and user guide

* Update heterograph.py

Co-authored-by: Ubuntu <ubuntu@ip-172-31-13-9.us-west-2.compute.internal>
2020-09-17 03:01:12 +08:00
Quan (Andy) Gan 76d66fd35d [Bug] Fix dtype mismatch in EdgeDataLoader on Windows (#2188)
* fix node and edge dataloader on windows

* fix distributed
2020-09-13 18:25:05 +08:00
Mufei Li 877101a96b [Doc] Update Doc for UDFs (#2131)
* Update Doc for UDFs

* add degree bucketing explanation

* Update udf.rst

* Update

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2020-09-13 04:57:06 +08:00
Jinjing Zhou 3234189b60 Fix hdfs error when writing big data (#2186)
* update dmlc-core to latest

* fix hdfs
2020-09-12 23:58:36 +08:00
Quan (Andy) Gan 4dae26a1fc [Doc] Fix typo in dataloading.negative_sampler (#2177)
* fix dataloading docs

* fix another
2020-09-11 19:39:48 +08:00
Quan (Andy) Gan 7d8522a29f [Bug] fix cumsum on an empty array with prepend_zero returning an empty array (#2179)
* fix cumsum

* udp

Co-authored-by: Zihao <expye@outlook.com>
2020-09-11 18:23:44 +08:00
Henry Kenlay 8a227bfa03 [Bugfix] fix TUDataset labelling issue (#2165) (#2173)
* [Bugfix] fix TUDataset labelling issue (#2165)

* [Bugfix] fix TUDataset labelling issue (dmlc#2165)

* update docstring according to discussion

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2020-09-11 17:05:24 +08:00
Xiangkun Hu a102db23bf [Dataset] CoraFull dataset remove redundant reference (#2172)
* PPIDataset

* Revert "PPIDataset"

This reverts commit 264bd0c960cfa698a7bb946dad132bf52c2d0c8a.

* Update gnn_benckmark.py
2020-09-11 15:15:06 +08:00
Quan (Andy) Gan 8f4594074b [Hotfix] Fix degree bucket edge ordering (#2176)
* fix degree bucket edge ordering

* unit test

* fix
2020-09-11 14:16:52 +08:00
Zihao Ye cbd55eb12f [performance] Batch DGLGraph in C++ end. (#2155)
* upd

* upd

* upd

* upd

* upd

* upd

* upd

* upd

* upd

* upd

* upd

* fix

* upd

* upd

* upd

* upd

* fix

* upd

Co-authored-by: VoVAllen <jz1749@nyu.edu>
2020-09-11 02:54:27 +08:00
Quan (Andy) Gan ac570c1dae [Bugfix] Fix flatten not wrapping unit graph (#2170)
* fix flatten not wrapping unit graph

* fix doc
2020-09-10 17:41:52 +08:00
Quan (Andy) Gan 5f44a4efce Fix #1453 again (#2169)
* Fix reincarnation of #1453

* fix
2020-09-10 15:43:21 +08:00
Quan (Andy) Gan b10b541c11 fix multiple docstring inconsistencies (#2167) 2020-09-10 11:03:34 +08:00
Zihao Ye 45b610c49a fix edge_softmax (#2160)
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2020-09-09 14:19:43 +08:00