文件历史

249 次代码提交

作者 SHA1 备注 提交日期
Zihao Ye e6bf54cdcb [Feature] Support aggregate multiple edge features in to_simple. (#2623)
* upd

* fix

* upd

* upd

* upd

* upd

* upd

* upd

* upd

* upd

* upd

* fix

* refactor

* upd test

* large feat_len or n in segment reduce

* lint
2021-02-21 10:05:42 +08:00
nv-dlasalle bc3a532f5e [Sampling] Implement dgl.to_block() for the GPU (#2339)
* Add start of to_block gpu implementation

* Pull in more changes from 0.4.2 cuda_to_block

* Move more code to IdArray

* Refactor DeviceNodeMapMaker

* Updates

* get compiling

* Integrate to_block

* Fix ID allocation

* Minor fixes

* Cleanup cuda calls to use cuda_common

* Reduce kernel calls

* Lint cleanup

* Expand documentation

* Remove unused function

* Rename variables for consistency

* Add doxygen comments

* Fix file extension

* Remove raw asynccopy for deviceapi

* Remove unused function

* Fix block/tile configuration

* Add cuda_device_common.cuh

* Add basic hashtable

* Migrate part of hashtable

* Refactor to use external hashtable

* Make functions members

* Format hash table functions

* Migrate duplicate filling

* Move last function over

* Refactor with cu file

* lint c++ code

* Move context check to C++ code

* Use macro switch

* Add missing files

* Update docstring

* update docs

* Move atomic functions

* Refactor hashtable

* Fix linting

* Expand docs

* Fix mismatched argument names

* Switch doxygen comments from using @param to \param

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-02-08 13:00:43 +08:00
Jinjing Zhou 0346b0aa80 [Test] Add test for Cython and force rebuild when run setup.py (#2635) 2021-02-05 22:38:46 +08:00
Zihao Ye fb4a0508c6 [bugfix] Solve the boundary issue in backward function of segment sum (#2610)
* upd

* trigger

* upd
2021-02-03 18:50:07 +08:00
Zihao Ye 0f9056edb9 [feature] Implement missing CUDA operators for COO format (part 1). (#2565)
* upd

* upd

* upd

* upd

* fix

* upd

* upd

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2021-01-25 16:15:22 +08:00
Quan (Andy) Gan 2288098ba6 [Feature] Converting a block to a graph and allow block construction (#2555)
* first commit

* more commit

* fix and docstring

* add to api reference

* address comments

* Update convert.py
2021-01-24 12:44:54 +08:00
Andrew 8c8c5b1dc1 [Bug fix] Stop erasing schemas of frames with no data rows (#2529)
* added option for batching empty data, fixes #2526

* added option for batching empty data, fixes #2526

* decreased line lengths

* removed trailing whitespace

* fixed wrong feature name

* [Bugfix] stop erasing schema of frames w/ no cols (fixes #2528)

* removed arg doc

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-01-17 10:32:47 +08:00
Andrew 0778766411 [Bug fix] [Feature] added option for batching empty data (#2527)
* added option for batching empty data, fixes #2526

* added option for batching empty data, fixes #2526

* decreased line lengths

* removed trailing whitespace

* fixed wrong feature name

* now default behavior when all graphs are empty

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-01-14 15:25:13 +08:00
Quan (Andy) Gan 2caac086c2 [Bug] send_and_recv and pull may write to wrong places (#2497)
* fix

* fix test
2021-01-06 17:18:37 +08:00
Mufei Li 0c2a2ea1e2 [Bugfix] Handle a Corner Case of Batching after Removing Nodes/Edges (#2465)
* Update

* Update

* Update

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2021-01-05 22:22:03 +08:00
Zihao Ye f960468f50 [feature] APIs for manually set batch_num_nodes and batch_num_edges (#2430)
* wip

* udp

* upd

* fix typo

* lint

* lint

* upd

* upd

* lint

* lint

* upd

* upd

* tftest

* fix

* fallback numpy

* fix tf

* docstring

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2021-01-04 23:43:58 +08:00
Minjie Wang d3c1645550 [Performance][Test] New low memory implementation for RGCN and related regression tests (#2468)
* WIP

* finish lowmem impl; benchmarking

* wip

* wip

* fix benchmarks

* fix bug in searchsorted

* update readme

* update

* lint and ogb dependency

* add flags to to_homogeneous

* fix docstring

* address comments
2021-01-04 22:23:59 +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
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
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 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
Quan (Andy) Gan 7b97298150 fix bug in flatten and is_unibipartite (#2279) 2020-10-17 12:11:24 +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
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 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
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
xiang song(charlie.song) c9c6171b9f Revert "[Feature] Basic utils to handle raw data features (#2102)" (#2147)
This reverts commit 33a8bb93bf.

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2020-09-08 10:33:59 +08:00
Quan (Andy) Gan ac5c79cdb8 [Bug] Fixes issues in feature pickling and transmission for subgraphs and blocks (#2139)
* bug fixes

* remove __deepcopy__; not sure what the behavior should be

* lint

* skip gpu test

* fix

* fix dist dataloader

* add comment

* remove assert
2020-09-03 12:48:59 +08:00
Quan (Andy) Gan 9eaace9216 rename create_format_ to create_formats_ (#2126) 2020-08-29 22:49:27 +08:00
Mufei Li 642bdbaf04 [Traversal] Fall back to CPU for Graph Traversal as a Workaround (#2115)
* workaround

* Allow Test for Traversal on GPU Graphs

Co-authored-by: Ubuntu <ubuntu@ip-172-31-1-5.us-west-2.compute.internal>
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2020-08-29 16:34:31 +08:00
Zihao Ye 3808dc950b [hotfix] Disable hypersparse memory optimization. (#2121)
* upd

* upd
2020-08-28 15:07:53 +08:00
xiang song(charlie.song) 33a8bb93bf [Feature] Basic utils to handle raw data features (#2102)
* add feature utils and add test for feature norm

* Add docstring and test

* upd

* dis able some test

* Update

* update doc string

* update

Co-authored-by: Ubuntu <ubuntu@ip-172-31-68-185.ec2.internal>
2020-08-27 11:36:39 +08:00
Quan (Andy) Gan 2ce426d956 [Bug] add_edges() crashes if the input tensor is empty (#2100)
* [Bug] add_edges() crashes if the input tensor is empty

* lint

* fix
2020-08-24 16:04:41 +08:00
Mufei Li 3af0e91c0e [Frame] Fix for Column Index Device (#2055)
* Update frame.py

* add unit test

* fix test

* fix

* fix

* fix

Co-authored-by: Quan Gan <coin2028@hotmail.com>
2020-08-19 16:10:52 +08:00
Mufei Li dacc7afeb2 [Convert] Reserve node/edge IDs in to_heterogeneous (#2045)
* Fix

* Fix

Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
2020-08-18 23:16:33 +08:00
Mufei Li 9088c63592 [Feature] Change the Default Value of transpose in DGLGraph.adj (#2046)
* Update adj

* Fix
2020-08-18 22:45:46 +08:00
Mufei Li 47c96dd114 [Test] Remove Test for metapath_random_walk (#2039)
* Update test_randomwalk.py

* Update test_randomwalk.py
2020-08-18 11:19:31 +08:00
Quan (Andy) Gan aa19df1bbb [Bugfix] Fix batching isolated node types (#2035)
* [Bugfix] Fix batching isolated ntypes

* fix test

* remove test.py

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2020-08-18 09:29:05 +08:00
Mufei Li be444e52d9 [Doc/Feature] Refactor, doc update and behavior fix for graphs (#1983)
* Update graph

* Fix for dgl.graph

* from_scipy

* Replace canonical_etypes with relations

* from_networkx

* Update for hetero_from_relations

* Roll back the change of canonical_etypes to relations

* heterograph

* bipartite

* Update doc

* Fix lint

* Fix lint

* Fix test cases

* Fix

* Fix

* Fix

* Fix

* Fix

* Fix

* Update

* Fix test

* Fix

* Update

* Use DGLError

* Update

* Update

* Update

* Update

* Fix

* Fix

* Fix

* Fix

* Fix

* Fix

* Fix

* Fix

* Update

* Fix

* Update

* Fix

* Fix

* Fix

* Update

* Fix

* Update

* Fix

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Fix

* Fix

* Update

* Update

* Update

* Update

* Update

* Update

* rewrite sanity checks

* delete unnecessary checks

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Fix

* Update

* Update

* Update

* Fix

* Fix

* Fix

* Update

* Fix

* Update

* Fix

* Fix

* Update

* Fix

* Update

* Fix

Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
Co-authored-by: Quan Gan <coin2028@hotmail.com>
2020-08-18 04:26:29 +08:00
Quan (Andy) Gan cd48435206 [Transform] Docstring and subframes (#1962)
* update knn graph docs

* more docs

* [Doc] transform module docstrings

* remove copy_ndata and copy_edata

* fix

* lint

* fix

* fix

* fix

* clean up docstrings

* fix docstring

* dtype specifications

* addresses comments

* fix

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
2020-08-13 10:41:43 +08:00
Zihao Ye 6294677f8a [hotfix] Set reduce results to all zero for nodes with zero in-degrees. (#2011) 2020-08-13 01:20:56 +08:00
Jinjing Zhou 5362908236 Fix save/load for old version files (#1994)
* load hetero

* fix

* more error message

* add warning

* add alias to dgl namespace

* fix tests
2020-08-12 23:45:21 +08:00
xiang song(charlie.song) 5e34ca8bb5 [Feature] Lazy copy ndata, edata to device (#1986)
* Lazy to device

* remove print

* lint

* Fix

* Fix

* Fix

* Fix

* Fix

* Fix

* Fix

* lint

* Fix

* Revert "Fix"

This reverts commit 615c9b8f80f5f6ee2ab43c849a22f0083deedf3b.

* Add test for frame lazy update

* disable tensorflow

* upd

Co-authored-by: Ubuntu <ubuntu@ip-172-31-51-214.ec2.internal>
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2020-08-12 13:04:47 +08:00
Quan (Andy) Gan f5eb80d221 [Feature] Edge DataLoader for edge classification & link prediction (#1828)
* clean commit

* oops forgot the most important files

* use einsum

* copy feature from frontier to block

* Revert "copy feature from frontier to block"

This reverts commit 5224ec963eb6a3ef1b6ab74d8ecbd44e4e42f285.

* temp fix

* unit test

* fix

* revert jtnn

* lint

* fix win64

* docstring fixes and doc indexing

* revert einsum in sparse bidecoder

* fix some examples

* lint

* fix due to some tediousness in remove_edges

* addresses comments

* fix

* more jtnn fixes

* fix
2020-08-11 18:00:58 +08:00
Da Zheng 729ff2ef38 Fix the performance issue of graph partitioning in new DGLGraph (#1934)
* fix perf.

* fix.

* accelerate metis.

* fix lint.

* use gklib.

* fix perf.

* fix.

* update metis.

* update launch script

* handle synchronized API.

* fix.

* fix example.

* fix dataloader.

* temp fix.

* temp fix omp.

* distinguish roles.

* initialize iterator of DistDataloader correctly.

* check the correctness of launch script.

* move feature copy to sampler.

* measure mem/network copy time.

* remove

* Revert "measure mem/network copy time."

This reverts commit 86cefdc14b7815fcf5aad6496af912dba48e4aa6.

* fix.

* fix

* fix.

* fix cmake.

* disable metis in windows.

* disable metis tests in windows.

* remove test for multigraph.

* fix test.

* fix.

* fix cmake.

* fix.

* revert.

Co-authored-by: Ubuntu <ubuntu@ip-172-31-19-115.us-west-2.compute.internal>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-19-1.us-west-2.compute.internal>
2020-08-09 18:33:05 -07:00
Zihao Ye 18bfec24cf [hotfix] Refactor edge softmax module (#1967)
* upd

* upd

* upd

* upd

* upd

* upd

* upd

* upd
2020-08-07 23:50:12 +08:00
Minjie Wang 22167f7203 [Refactor] Enable new kernel in all message passing APIs (#1953)
* WIP: frame refactor

* new frame

* simple update_all builtin

* move all subgraph routines into the same file

* sddmm & spmm schedule; node & edge udf

* degree bucketing

* some tricky 0deg corner cases

* bug in frame append

* merge test_hetero_basics and test_basics

* some code rearange

* fix test_heterograph

* add mean spmm

* enable all builtin combinations

* pass gpu test

* pass pytorch tests

* wip

* fix some pt debugging codes

* fix bug in mxnet backward

* pass all mxnet utests

* passed tf tests

* docstring

* lint

* lint

* fix broadcasting bugs

* add warning and clamp for mean reducer

* add test for zero-degree mean

* address comments

* lint

* small fix
2020-08-07 15:40:25 +08:00
xiang song(charlie.song) 35c9473ba5 [HotFix] Fix add_reverse_edges (#1960)
* Hot fix

* more test

Co-authored-by: Ubuntu <ubuntu@ip-172-31-51-214.ec2.internal>
2020-08-07 13:07:06 +08:00
xiang song(charlie.song) 879e4ae5f2 [Transform] Add add_reverse_edges (#1936)
* Add add_reverse_edges and is_simple_graph
Update to_bidirected

* remove is_simple_graph

* Fix lint

* to bidirected only support cpu

* support copy ndata

* upd

Co-authored-by: Ubuntu <ubuntu@ip-172-31-51-214.ec2.internal>
2020-08-05 15:09:55 +08:00
Zihao Ye d2ccd218cf [hotfix] Allow creating graph from edges with restrict formats excluding coo. (#1926)
* upd

* upd

* upd

* upd
2020-08-04 00:33:36 +08:00
Tong He b5be4f4bbf [Dataset] Update dataset hash with deterministic function (#1919)
* update hash with deterministic function

* update docstring

* add test on determination
2020-08-03 16:17:00 +08:00
Zihao Ye 03f251ced1 upd (#1905) 2020-07-31 20:13:24 +08:00