3564fdc5f1
[Bugfix][Model] fix treelstm model ( #274 )
...
* fix bug after moving batcher out of dgl.data
* disable mx utest
2018-12-06 23:52:50 -05:00
7aa494b3bc
[Bugfix][Doc] explain the batch dimension in the doc and many fix ( #266 )
...
* add more unit tests for mxnet.
* fix.
* explain the batch dimension.
* update doc.
* disable unit tests on DFS.
* fix graph traversal.
2018-12-06 00:48:57 -05:00
063ed69cdf
[BUGFIX] fix a bug in creating immutable graph index. ( #251 )
...
* fix a bug in creating immutable graph index.
* fix for new changes in the backend API.
* fix for creating immutable graph index from coo matrix.
* retrigger
2018-12-05 17:51:50 -05:00
899d125bb1
Add tests for MXNet and throw NotImplementedError if not implemented ( #258 )
...
* add more unit tests for mxnet.
* fix.
2018-12-05 10:41:33 -08:00
7c7cc7e0c2
[sampler] Adjust the sampler API for the future extension. ( #243 )
...
* return seed ids.
* fix tests.
* implement.
2018-12-05 10:05:10 -08:00
21255b6528
[Bugfix] tolist and dependencies in dgl.data ( #239 )
...
* change Index.tolist -> Index.tonumpy; fix bug in traversal; remove dependencies in data
* fix import
* fix __all__ and some docstring
2018-12-05 12:20:59 -05:00
eafcb7e7f5
[Bugfix][MXNet] Fix edge order and builtin max bug in mx ( #247 )
...
* Fix edge order and builtin max bug in mx
* fix as requested
2018-12-05 01:32:54 -08:00
2c170a8c17
[Graph][Bugfix] Fix the API of map_to_subgraph_nid ( #226 )
...
* correct vid mapping API.
* fix sse.
2018-12-03 17:16:35 -05:00
8918cce0b5
[MXNet] add unit tests for mxnet ( #227 )
...
[MXNet] add unit tests for mxnet
2018-12-03 11:04:11 -05:00
b1eeb93449
[Bugfix][Runtime] Zero degree behaviors ( #177 )
...
* fix recv nodes are all 0deg; fix hybriddict does not through keyerror properly
* fallback to apply_nodes when all nodes are 0deg; WIP on pull spmv 0deg
* new 0deg behavior
* new 0deg behavior
* update mx utest for pull-0deg
* fix mx
* fix mx
* get rid of unnecessary sort-n-unique
2018-11-25 22:33:08 -05:00
deb653f8dc
[Runtime] Scheduler and Executor ( #140 )
...
* executor api
* draft executor interface
* WIP
* revert changes to avoid conflict with api change
* core scheduling logic
* WIP: build graph adj
* incidence matrix for in edges
* support incidence matrix for partial recv nodes
* improve
* build adjmat in scheduler
* graph store
* get degree bucketing schedule
* connect to c++ degree bucketing
* conceptual executor creation code
* executor comments
* fix
* more executor comments
* WIP: full send_and_recv schedule
* most schedulers
* simplify scheduler
* executors
* runtime
* builtin function base class
* adj indices and shape
* completely refactor scheduler
* rename and move bundled out to function.py
* use_edge_feature in msg func
* rewrite scheduler
* node edge executor
* connect with graph api
* handle zero degree
* misc
* fix test cases
* fix a good many bugs...
* remove old scheduler
* push and pull
* fix send recv
* c++ lint
* fix batched send recv
* hot fix for mxnet
* typo
* write back executor
* apply node edge
* clean up, doc string
* fix as requested
* refactor
* fix
* WIP
* WIP
* ir draft
* more on ir
* WIP: spmv schedule
* WIP
* recv schedule
* refactor
* WIP
* snr degree bucketing
* snr scheduler
* move prog to graph.py; rename
* unittest for send/recv
* remove some legacy codes
* WIP: update_all
* pass test_basics
* passed all current utests
* more utests; fix mx utest
* WIP: fixing zero deg initial value
* some tests
* fix 0deg problem
* fix mx
* fix mx
* some notes
* fix as requested
2018-11-22 12:15:50 -05:00
440aecee8f
[Test] add tests for built-in functions ( #159 )
...
* add spmv tests on mxnet.
* test backward.
* add tests for pull.
* use mxnet backend.
* address comment.
* add comments.
2018-11-20 23:39:54 -05:00
2389df8155
[API] Subgraph sampling API. ( #138 )
...
* add neighbor sampler.
* fix sampler.
* handle non-existing nodes differently in MapParentIdToSubgraphId.
* fix sampling tests.
* parallelize sampling.
* add more tests.
* fix a bug in subgraph loader.
* get number of sampled nodes directly.
* load mxnet CSR to DGLGraph directly.
* create graph from mxnet csr array directly.
* use subgraph loader in SSE.
* fix a bug in sampling.
* output framework tensor.
* fix.
* remove unnecessary code.
* fix tests.
* add SSE README.
* move to contrib.
* address comments.
* add tests.
2018-11-14 13:51:27 -08:00
048f6d7a30
[GraphIndex] refactor graph caching ( #150 )
...
* refactor graph caching
* fix mx test
* fix typo
2018-11-13 22:12:38 -05:00
bd0e4fa0b3
[MXNet][API] move to the new API ( #123 )
...
* move gat to the new api.
* fix gcn.
* update sse.
* fix dgl core.
* update sse.
* fix small bugs in dgl core.
* fix mxnet tests.
* retrigger
* address comments and fix more bugs.
* fix
* fix tests.
2018-11-08 10:00:47 -08:00
7241a9c008
[Backend] backend interface ( #109 )
...
* backend interface
* small fix
* more comments to the data type dict
* WIP
* convert_to and narrow
* WIP
* pytorch and numpy backend; WIP on mxnet backend
* mxnet backend
* narrow
* Fix all usages
* fix for mx
* fix for mx
* fix mx
* fix mx
* fix mx
* fix mx
* fix mx
* fix mx
* fix mx
* revert jenkins
* add sparse_matrix api
* sparse matrix api
* some fixme
* Fix as requested
2018-11-04 21:42:39 -05:00
68ec624782
[API][Doc] API change & basic tutorials ( #113 )
...
* Add SH tutorials
* setup sphinx-gallery; work on graph tutorial
* draft dglgraph tutorial
* update readme to include document url
* rm obsolete file
* Draft the message passing tutorial
* Capsule code (#102 )
* add capsule example
* clean code
* better naming
* better naming
* [GCN]tutorial scaffold
* fix capsule example code
* remove previous capsule example code
* graph struc edit
* modified: 2_graph.py
* update doc of capsule
* update capsule docs
* update capsule docs
* add msg passing prime
* GCN-GAT tutorial Section 1 and 2
* comment for API improvement
* section 3
* Tutorial API change (#115 )
* change the API as discusses; toy example
* enable the new set/get syntax
* fixed pytorch utest
* fixed gcn example
* fixed gat example
* fixed mx utests
* fix mx utest
* delete apply edges; add utest for update_edges
* small change on toy example
* fix utest
* fix out in degrees bug
* update pagerank example and add it to CI
* add delitem for dataview
* make edges() return form that is compatible with send/update_edges etc
* fix index bug when the given data is one-int-tensor
* fix doc
2018-11-02 17:19:14 -04:00
9e9a9488f1
[GraphIndex] Immutable graph index using MXNet CSRArray ( #86 )
...
* add immutable graph index.
* update ImmutableGraphIndex.
* add benchmarks of subgraph generation.
* fix node_subgraphs.
* use scipy synthetic data in immutable graph.
* create immutable_graph.
* use spmv in gcn.
* fix immutable graph index.
* update graph index.
* create ImmutableSubgraphIndex.
* return subgraph node and edge mapping.
* fix benchmark.
* add mapping to subgraph nid.
* test in_edges and out_edges.
* update benchmark.
* Fix immutable subgraph.
* complete in_edges and out_edges.
* fix benchmarks.
* add test on node_subgraphs
* add sse.
* construct subgraphs in parallel.
* add in/out degree(s).
* make immutable graph index backend-specific.
* implement has_edge(s) and edge_id(s).
* Revert "use spmv in gcn."
This reverts commit 9cfed5f5fb3dd2ed9b98745348b1c0e9731ed7f7.
* implement node_subgraphs in GraphIndex.
* fix sse.
* address comments.
* address comments.
* Update mxnet docker.
2018-10-26 11:28:25 -07:00
0d6cd30be3
[CI] Enable mxnet CI. ( #105 )
...
* fix tests.
* enable mxnet CI.
* add mxnet tests.
* update jenkins.
* update mxnet docker image.
2018-10-26 12:04:41 -04:00
4af3f8bcc8
update the MXNet backend. ( #89 )
...
* update mxnet.
* add get_tvmtype.
* remove undefined test.
2018-10-18 12:49:13 -07:00
5567f4a4d9
[BACKEND] Add MXNet backend. ( #77 )
...
* support mxnet.
* add mxnet version of GCN.
* rename mxnet.nd as F.
* add mxnet GAT.
* enable GPU for GCN.
* fix MXNet GCN train.
* Use adam to optimize GAT
* support more operators.
* support sparse arrays.
* update mxnet backend.
* support index_copy.
* remove NN.
* update mxnet backend.
* temp check in.
* fix data conversion.
* add test.
* clean up mxnet backend.
* update mxnet examples.
* Revert "remove NN."
This reverts commit d815d9a0ec619f9ce9099c48cd35db9d8e947483.
* temp disable MXNet version of NN.
2018-10-14 11:15:48 -04:00
Minjie Wang