文件历史

2100 次代码提交

作者 SHA1 备注 提交日期
zengxy 84702916d3 [BugFix] Some negative layer_id/block_id are not converted to the actual one. (#584)
* [BugFix] Some negative layer_id/block_id are not converted to the actual one.

* Add some tests for negatives block/layer ids
2019-06-04 12:19:43 -07:00
Zihao Ye a3f10c2138 [bugfix] Fixed bug in from_networkx (#603)
* minor fix

* small fix
2019-06-04 11:55:21 +08:00
Minjie Wang 01a4cc5b7b [Graph] Add API to convert graph to simple graph (#587)
* to simple

* WIP: multigraph flag

* graph index refactor; pass basic testing

* graph index refactor; pass basic testing

* fix bug in to_simple; pass torch test

* fix mx utest

* fix example

* fix lint

* fix ci

* poke ci

* poke ci

* WIP

* poke ci

* poke ci

* poke ci

* change ci workspace

* poke ci

* poke ci

* poke ci

* poke ci

* delete ci

* use enum for multigraph flag
2019-06-01 23:32:46 -04:00
Da Zheng 372203f04a [Test] Fix tests in test_shared_mem_store. (#588)
* fix test.

* better assert.

* more asserts.

* print to stderr.

* destroy g.

* fix tests.

* add timeout in sync_barrier.

* test _sync_barrier.

* fix.

* avoid printing messages.

* fix test.

* fix test.

* fix.
2019-06-01 21:18:01 -04:00
VoVAllen 16af6ebd3b [Test] Add cpp test to CI (#567)
* add cpptest to ci

* try windows

* Update Jenkinsfile

* Update Jenkinsfile

* Update Jenkinsfile

* Update task_cpp_unit_test.bat

* Change localhost to 127.0.0.1

* windows socket gtest

* lint

* Update task_cpp_unit_test.bat

* fix batch script

* export all symbols

* Update Jenkinsfile

* compile once

* try mxnet compact
2019-05-30 11:01:16 -04:00
Mufei Li 306bb3f9ad [Bugfix] Fix minigc (#585) 2019-05-29 21:59:03 -04:00
Minjie Wang 21e172d076 [Bugfix] fix numpy integer bug (#580) 2019-05-29 21:48:36 -04:00
Da Zheng 40dc18595b [Test] fix CI. (#586)
* disable unit test in mxnet tutorial.

* retry socket connection.

* roll back to set_np_compat

* try to fix multi-processing test hangs when it fails.

* fix test.

* fix.
2019-05-29 21:40:16 -04:00
Lingfan Yu ada3786c57 [Docker]Update MXNet CI Image (#582)
* use mxnet 20190528 build

* fix mxnet api name
2019-05-28 18:04:54 -07:00
Da Zheng bd2d2c75b5 [BUGFIX] fix for MXNet 1.5. (#552)
* remove clone.

* turn on numpy compatible.

* Revert "remove clone."

This reverts commit 17bbf76ed72ff178df6b3f35addc428048672457.
2019-05-28 14:42:44 -07:00
askliar b3097725f2 [BugFix] Fix error when bfs_level = 0 in Entity Classification with RGCN (#559) 2019-05-25 17:38:22 -07:00
Da Zheng 8f378d90ad [BUGFIX] fix bugs for running GCN on giant graphs. (#561)
* load mxnet csr.

* enable load large csr.

* fix

* fix.

* fix int overflow.

* fix test.
2019-05-25 11:09:33 -07:00
Da Zheng 14af8402f0 [Perf] lazily create msg_index. (#563)
* lazily create msg_index.

* update test.
2019-05-24 13:41:03 -07:00
Da Zheng 924efc6520 [Perf] Improve performance of graph store. (#554)
* fix.

* use inplace.

* move to shared memory graph store.

* fix.

* add more unit tests.

* fix.

* fix test.

* fix test.

* disable test.

* fix.
2019-05-23 12:23:31 -07:00
Minjie Wang 605b51857e [Refactor] Immutable graph index (#543)
* WIP

* header

* WIP .cc

* WIP

* transpose

* wip

* immutable graph .h and .cc

* WIP: nodeflow.cc

* compile

* remove all tmp dl managed ctx; they caused refcount issue

* one simple test

* WIP: testing

* test_graph

* fix graph index

* fix bug in sampler; pass pytorch utest

* WIP on mxnet

* fix lint

* fix mxnet unittest w/ unfortunate workaround

* fix msvc

* fix lint

* SliceRows and test_nodeflow

* resolve reviews

* resolve reviews

* try fix win ci

* try fix win ci

* poke win ci again

* poke

* lazy multigraph flag; stackoverflow error

* revert node subgraph test

* lazy object

* try fix win build

* try fix win build

* poke ci

* fix build script

* fix compile

* add a todo

* fix reviews

* fix compile
2019-05-21 14:28:06 -04:00
Da Zheng b2b8be25bb [API] update graph store API. (#549)
* add init_ndata and init_edata in DGLGraph.

* adjust SharedMemoryGraph API.

* print warning.

* fix comment.

* update example

* fix.

* fix examples.

* add unit tests.

* add comments.
2019-05-21 03:27:05 -07:00
Da Zheng cdfca992e1 [BUGFix] Improve multi-processing training (#526)
* fix.

* add comment.

* remove.

* temp fix.

* initialize for shared memory.

* fix graphsage.

* fix gcn.

* add more unit tests.

* add more tests.

* avoid creating shared-memory exclusively.

* redefine remote initializer.

* improve initializer.

* fix unit test.

* fix lint.

* fix lint.

* initialize data in the graph store server properly.

* fix test.

* fix test.

* fix test.

* small fix.

* add comments.

* cleanup server.

* test graph store with a random port.

* print.

* print to stderr.

* test1

* test2

* remove comment.

* adjust the initializer signature.
2019-05-20 14:23:19 -07:00
VoVAllen aa129522e0 [Bug Fix] Fix inplace op at backend (#546)
* Fix inplace operation

* fix line seprator
2019-05-19 23:43:54 -04:00
zengxy d533e9ba62 [BugFix] Fix getting src and dst id of ALL edges in NodeFlow.apply_block (#515) 2019-05-16 18:08:43 -07:00
Da Zheng 3a1392e641 [Model] add multiprocessing training with sampling. (#484)
* reorganize sampling code.

* add multi-process training.

* speed up gcn_cv

* fix graphsage_cv.

* add new API in graph store.

* update barrier impl.

* support both local and distributed training.

* fix multiprocess train.

* fix.

* fix barrier.

* add script for loading data.

* multiprocessing sampling.

* accel training.

* replace pull with spmv for speedup.

* nodeflow copy from parent with context.

* enable GPU.

* fix a bug in graph store.

* enable multi-GPU training.

* fix lint.

* add comments.

* rename to run_store_server.py

* fix gcn_cv.

* fix a minor bug in sampler.

* handle error better in graph store.

* improve graphsage_cv for distributed mode.

* update README.

* fix.

* update.
2019-05-06 23:51:56 -07:00
Da Zheng 8c79885d8b [BUGFIX] fix the error message in the data loader. (#509)
* fix.

* Change to required=True

* Update description

* Change back
2019-04-26 09:52:45 +09:00
Zhengwei ad9da36add [Model] Add DGI Model (#501) 2019-04-22 11:37:06 +08:00
Chao Ma fe7d5e9b8b [Sampler] Change Distributed Sampler API (#499)
* Change Distributed Sampler API

* fix lint

* fix lint

* update demo

* update

* update

* update

* update demo

* update demo
2019-04-22 09:41:47 +08:00
kitaev-chen a3febc061b [Model] Add GIN Model (#471)
* add gin model

* convert dataset.py to data_ont_the_fly way and put it into dgl.data module

* convert dataset.py to data_ont_the_fly way and put it into dgl.data module
python code checked

* modified document and reference TUDataset; checked python part and bypass cpp part due to error

* change tensor to numpy in dataset and transform in collate@Dataloader

* Change minor format issue

Change minor format issue

* moved logging; adjusted tqdm etc
2019-04-17 14:57:07 +08:00
Quan (Andy) Gan fb6af16f83 [BUG] Fix crash during Cython build (#500) 2019-04-17 14:18:22 +08:00
Chao Ma b89dcce16c [RPC] Refactoring networking APIs (#496)
* Refactoring network API

* update demo

* update

* update demo

* update demo

* add num_sender

* update

* fix lint

* fix lint

* fix lint

* update
2019-04-16 16:33:29 +08:00
VoVAllen c69dbbb736 [Bug Fix] Fix TUDataset Processing bug (#489)
* add graph classification dataset

* add node label

* add TUDataset

* Modify to consistent with Qi Huang's implementation

* add docs

* Add docs

* Fix change of environment variable

* Update tu.py

* Update tu.py

* Fix error when add node with np.int64

* Fix processing bug
2019-04-11 19:30:50 +08:00
VoVAllen 00fc680aae [Dataset] Add TUDataset (#473)
* add graph classification dataset

* add node label

* add TUDataset

* Modify to consistent with Qi Huang's implementation

* add docs

* Add docs

* Fix change of environment variable

* Update tu.py

* Update tu.py

* Fix error when add node with np.int64
2019-04-10 09:37:53 -04:00
Da Zheng bfdd1eaa44 [Feature] Create shared memory graph store. (#468)
* accelerate gcn_ns.

* add timing.

* run infer with whole graph.

* distributed gcn_ns.

* reconstruct gcn_ns.

* minor fix.

* change graphsage_cv for numa.

* fix #OMP threads.

* accelerate graphsage_cv.

* fix a weird bug.

* add profiler in graphsage_cv.

* accelerate graphsage_cv.

manually aggregate neighbors' embeddings with pull.

* load csr directly in gcn_ns_sc.

* parallel sort for graph index.

* Revert "parallel sort for graph index."

This reverts commit 86fe2c7117fe5e56b0d481b39849c258b166945b.

* run gcn_ns_sc on GPUs.

* acc gcn_cv_sc.

* change gcn_cv for numa.

* fix gcn_cv to use numa and gpu.

* improve graphsage_cv to use numa and gpu.

* improve gcn_ns.

* improve graphsage_cv.

* init shared memory graph store.

* fix.

* enable init ndata.

* improve tests.

* add bidirectional communication.

* link to rt.

* fix compilation error.

* fix shared memory init.

* use MessageQueue for inter-process communication.

* reconstruct immutable graph csr.

* fix gcn.

* load csr to shared memory.

* fix minor bugs.

* add comments.

* refactor SharedMemory.

* fix bugs in ImmutableGraph.

* create CSR graph from shared memory.

* add more test for loading a csr graph.

* terminate graph store properly.

* allow initializing ndata in the graph store server.

* use RPC for inter-process communication.

* a script for loading a graph.

* allow customizing port.

* list all ndata and edata.

* support dtype.

* reorganize SharedMemoryGraphStore.

* fix ndata shape.

* reconstruct gcn_ns.

* print info.

* set omp in gcn_ns.

* reset sampling examples.

* fix lint.

* fix lint.

* reset gcn.

* disable shared memory in windows.

* fix.

* fix.

* reset changes.

* revert nodeflow changes.

* fix cmake.

* fix test.

* fix test.

* fix test.

* fix test.

* add comments.

* fix test.

* move vector out.

* fix lint.

* fix lint.

* move SharedMemory.

* update cmake.

* update comment.

* fix comments.

* Revert "update cmake."

This reverts commit 592445e37077f70a6e3f2e5245f9a3d086b04f3b.

* update cmake.

* add comments.

* rename.

* change the comment.

* fix a bug.

* rename.

* add comments.

* add comments.

* add init_edata.

* rewrite memory alloc.

* move vector to CSR.

* fix.

* init data.

* Revert "init data."

This reverts commit 2b217b9553911b7dd84a9f1d9b68430b5aa18e23.

* init data.

* init new columns correctly.
2019-04-08 12:09:02 -07:00
Chao Ma 8058f1c579 [Fix] Update inner API of distributed sampler (#478)
* update inner API of distributed sampler

* update
2019-04-05 10:00:36 +08:00
Chao Ma da3ab84c3c Add demo for distributed sampler (#474)
* add C++ rpc infrastructure and distributed sampler

* update

* update lint

* update lint

* update lint

* update

* update

* update

* updare

* update

* update

* update

* update serialize and unittest

* update serialize

* lint

* update

* update

* update

* update

* update

* update

* update unittest

* put Finalize() to __del__

* update unittest

* update

* delete buffer in Finalize

* update unittest

* update unittest

* update unittest

* update unittest

* update

* update

* fix small bug

* windows socket impl

* update API

* fix bug in serialize

* fix bug in serialzie

* set parent graph

* update

* update

* update

* update

* update

* update

* fix lint

* fix lint

* fix

* fix windows compilation error

* fix windows error

* change API to lower-case

* update test

* fix typo

* update

* add SamplerPool

* add SamplerPool

* update

* update test

* update

* update

* update

* update

* add example

* update

* update

* add distributed sampler demo

* add index

* update demo of distributed sampler

* fix lower-case

* print subg index

* update README.md

* update

* remove --gpu args
2019-04-03 16:09:59 -07:00
Chao Ma 5d494c6207 [RPC] add C++ RPC infrastructure and distributed sampler (#465)
* add C++ rpc infrastructure and distributed sampler

* update

* update lint

* update lint

* update lint

* update

* update

* update

* updare

* update

* update

* update

* update serialize and unittest

* update serialize

* lint

* update

* update

* update

* update

* update

* update

* update unittest

* put Finalize() to __del__

* update unittest

* update

* delete buffer in Finalize

* update unittest

* update unittest

* update unittest

* update unittest

* update

* update

* fix small bug

* windows socket impl

* update API

* fix bug in serialize

* fix bug in serialzie

* set parent graph

* update

* update

* update

* update

* update

* update

* fix lint

* fix lint

* fix

* fix windows compilation error

* fix windows error

* change API to lower-case

* update test

* fix typo

* update

* add SamplerPool

* add SamplerPool

* update

* update test

* update

* update

* update

* update

* add example

* update

* update
2019-03-29 16:48:45 +08:00
Quan (Andy) Gan 6066fee935 [Model][Feature] PinSage & Random Walk with Restart (#453)
* random walk traces generation

* remove outdated comments

* oops put in the wrong place

* explicit inline

* moving rand_r to util

* pinsage-like model on movielens

* the code runs now

* support cuda

* using readonly graph

* moving random walk to public function

* per-thread seed and openmp support

* pinsage-like model on movielens

* the code runs now

* support cuda

* using readonly graph

* using C random walk

* removing profile decorators

* param initialization

* no grad

* leaky relu fixes everything

* train and save

* WIP

* WIP

* WIP

* seems to work

* evaluation output

* swapping order of val/test and train

* debug

* hyperparam tuning

* prior/training dataset split changes

* random walk reorg

* random walk with restart

* signed comparison fix

* migrating random walk to nodeflow

* Revert "migrating random walk to nodeflow"

This reverts commit f2565347cced7c912a58a529b257c033d9f375b7.

* add README and remove dataset

* new endpoint

* lint

* lint x2

* oops forgot test

* including bpr - better for baseline

* addressing fixes

* throwing random walks out from SamplerOp class

* forgot to move RandomWalk; why did this even work?

* removing legacy garbage

* add todo

* address comments

* stupid bug fix

* call ndarrayvector converter to handle traces
2019-03-29 13:01:18 +08:00
Quan (Andy) Gan 341dbea0a1 [Bug] Fix unused prefetch switch for NeighborSampler and LayerSampler (#457)
* forgot prefetch switch

* yet another stupid bug
2019-03-19 14:57:10 +08:00
Minjie Wang 33b9c38389 [Bugfix] fix bug in pickling (#456)
* fix bug in pickling

* fix lint
2019-03-18 23:48:20 -04:00
Lingfan Yu 6f603bbf8f [BugFix] Fix performance bug in EdgeSoftmax (#452)
* fix performance bug in EdgeSoftmax

* minor
2019-03-18 16:22:38 -04:00
Quan (Andy) Gan 681e5217dd [Refactor] Sampler code refactor (#454)
* refactored sampler code

* docstring

* fix tutorial
2019-03-18 10:41:35 -04:00
Lingfan Yu eb1acecd0d [BugFix] Handle case when calling send with builtin message function (#449)
* fix send with builtin bug

* test cases

* remove todo comment
2019-03-13 14:49:50 -04:00
Zihao Ye d1d580ecc1 [feature & bugfix] adjacency_matrix_scipy, wrong order of edges in immutable graph. (#437)
* fix rgcn tutorial

* small fix

* upd

* findedge/s

* upd

* upd

* upd

* upd

* add test

* remove redundancy

* upd

* upd

* upd

* upd

* add edge_subgraph

* explicit cast

* add test immutable subg

* reformat

* reformat

* fix bug

* upd

* hotfix

* subgraph docs

* fix adj mat

* fix edges() order

* add test

* revert

* upd

* fix

* upd
2019-03-12 23:57:13 -04:00
mc-robinson 13272628b5 [Doc] added index [0] to torch.max (#443) 2019-03-08 18:29:16 -05:00
Minjie Wang d3cd60142c [Build] fix setup script to allow cython ext (#444) 2019-03-08 18:23:38 -05:00
Minjie Wang 2ade0ef96d [Hotfix] fix cython bug (#442)
* fix cython bug

* fix

* fix
2019-03-08 17:07:17 -05:00
Minjie Wang cd4f0a2bfc [Release] update version (#440)
* update version

* update news
2019-03-08 11:12:16 -05:00
Da Zheng 8651be54c2 [Perf] Accelerate block_compute when all nodes are invoked. (#434)
* refactor.

* accelerate update_all in nodeflow.

* fix.

* refactor.

* fix lint.

* fix lint.

* reorganize.

* reorg.

* remove.

* add doc.

* impl block_incidence_matrix

* fix lint.

* fix.

* simple fix.

* fix test.

* fix interface.

* fix eid.

* fix comments.
2019-03-07 13:39:39 -08:00
Minjie Wang ca2a7e1ca1 [Refactor] Nodeflow, sampling, CAPI (#430)
* enable cython

* add helper function and data structure for void_p vector return

* move sampler from graph index to contrib.sampling

* WIP

* WIP

* refactor layer sampling

* pass tests

* fix lint

* fix graphsage

* remove comments

* pickle test

* fix comments

* update dev guide for cython build
2019-03-05 14:07:22 -05:00
Lingfan Yu d3c24cc25b [BugFix] Fix bug in RGCN data processing and use index_select to improve speed (#429)
* use index_select instead of __getitem__

* fix bug in dataset processing

* fix edge_type shape bug

* comments
2019-03-04 10:52:21 -05:00
GaiYu0 fb4246e564 [Feature]Uniform layer-wise sampler (#416)
* migrate to node-flow

* uniform layer sampler test cases

* more test cases

* documentations

* fix lint errors

* fix lint errors

* fix lint errors

* iota

* add asnumpy

* requested changes

* fix indptr error

* fix lint errors

* requested changes & fix lint errors

* fix lint errors

* fix LayerSampler unit test
2019-03-02 16:37:18 -08:00
Minjie Wang a88f351174 [Doc] update doc string (#426) 2019-03-02 14:00:52 -05:00
Lingfan Yu 3cc32a9775 [Feature] Reduce messages with scatter_add in PyTorch (#427)
* implement pytorch spmm with gather and scatter add

* fix

* replace torch take with index_select

* comments

* comment about pytorch __getitem__ operator pitfall

* typo
2019-03-01 22:51:51 -05:00
Minjie Wang fe44ffe5dc [Doc] fix equation in nn.conv (#425) 2019-03-01 15:56:08 -05:00