Rhett Ying
cc83b49e01
[Bug] check dtype before convert to gk ( #3414 )
2021-11-04 06:13:19 +00:00
Rhett Ying
1c9274f168
[BugFix] initialize data if null when converting from row sorted coo to csr ( #3360 )
2021-11-04 06:08:45 +00:00
Mufei Li
f0fafa2062
[Bug fix] Fix batch information with remove_nodes/edges applied ( #3119 )
...
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Fix lint
* Fix
* Update
* Fix test cases
* Fix
* add docstrings
* Update
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-17.us-west-2.compute.internal >
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com >
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com >
2021-07-13 16:27:31 +08:00
Rhett-Ying
175f53decf
[Feature] enable edge reorder in dgl.reorder_graph() ( #3113 )
...
* [Feature] enable edge reorder in dgl.reorder_graph()
* refine doc string
* refine doc string for dgl.reorder_graph
* refine doc string further
2021-07-12 14:11:46 +08:00
Rhett-Ying
4307fe8895
[Feature] Add dgl.reorder() to re-order graph according to specified … ( #3063 )
...
* [Feature] Add dgl.reorder() to re-order graph according to specified strategy
* fix unit test failure for metis reorder
* fix unit test failure on mxnet_cpu
* refine unit test for dgl.reorder
* fix unit test failure on mxnet
* fix array_equal error for mxnet unit test
* fix unit test failure for mxnet
* convert metis output to numpy array explicitly
Co-authored-by: Tong He <hetong007@gmail.com >
2021-06-30 13:29:08 +08:00
Da Zheng
6e7f19f27b
[Distributed] add distributed in-degree and out-degree. ( #2918 )
...
* add distributed in-degree and out-degree.
* update comments.
* fix a bug.
* add tests.
* add tests.
* fix a bug.
* fix docstring.
* update doc.
* fix
* fix.
Co-authored-by: Zheng <dzzhen@3c22fba32af5.ant.amazon.com >
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com >
2021-05-18 19:46:09 +08:00
Da Zheng
d76af4d4c1
[Distributed] Return the ID mapping in graph partitioning. ( #2857 )
...
* return mapping.
* support heterogeneous graph.
* more test.
* fix lint.
* fix for diff backends.
* fix.
* fix.
Co-authored-by: Zheng <dzzhen@3c22fba32af5.ant.amazon.com >
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com >
2021-04-22 17:51:42 +08:00
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
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
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
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
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
Minjie Wang
44089c8b4d
[Refactor][Graph] Merge DGLGraph and DGLHeteroGraph ( #1862 )
...
* Merge
* [Graph][CUDA] Graph on GPU and many refactoring (#1791 )
* change edge_ids behavior and C++ impl
* fix unittests; remove utils.Index in edge_id
* pass mx and th tests
* pass tf test
* add aten::Scatter_
* Add nonzero; impl CSRGetDataAndIndices/CSRSliceMatrix
* CSRGetData and CSRGetDataAndIndices passed tests
* CSRSliceMatrix basic tests
* fix bug in empty slice
* CUDA CSRHasDuplicate
* has_node; has_edge_between
* predecessors, successors
* deprecate send/recv; fix send_and_recv
* deprecate send/recv; fix send_and_recv
* in_edges; out_edges; all_edges; apply_edges
* in deg/out deg
* subgraph/edge_subgraph
* adj
* in_subgraph/out_subgraph
* sample neighbors
* set/get_n/e_repr
* wip: working on refactoring all idtypes
* pass ndata/edata tests on gpu
* fix
* stash
* workaround nonzero issue
* stash
* nx conversion
* test_hetero_basics except update routines
* test_update_routines
* test_hetero_basics for pytorch
* more fixes
* WIP: flatten graph
* wip: flatten
* test_flatten
* test_to_device
* fix bug in to_homo
* fix bug in CSRSliceMatrix
* pass subgraph test
* fix send_and_recv
* fix filter
* test_heterograph
* passed all pytorch tests
* fix mx unittest
* fix pytorch test_nn
* fix all unittests for PyTorch
* passed all mxnet tests
* lint
* fix tf nn test
* pass all tf tests
* lint
* lint
* change deprecation
* try fix compile
* lint
* update METIDS
* fix utest
* fix
* fix utests
* try debug
* revert
* small fix
* fix utests
* upd
* upd
* upd
* fix
* upd
* upd
* upd
* upd
* upd
* trigger
* +1s
* [kernel] Use heterograph index instead of unitgraph index (#1813 )
* upd
* upd
* upd
* fix
* upd
* upd
* upd
* upd
* upd
* trigger
* +1s
* [Graph] Mutation for Heterograph (#1818 )
* mutation add_nodes and add_edges
* Add support for remove_edges, remove_nodes, add_selfloop, remove_selfloop
* Fix
Co-authored-by: Ubuntu <ubuntu@ip-172-31-51-214.ec2.internal >
* upd
* upd
* upd
* fix
* [Transfom] Mutable transform (#1833 )
* add nodesy
* All three
* Fix
* lint
* Add some test case
* Fix
* Fix
* Fix
* Fix
* Fix
* Fix
* fix
* triger
* Fix
* fix
Co-authored-by: Ubuntu <ubuntu@ip-172-31-51-214.ec2.internal >
* [Graph] Migrate Batch & Readout module to heterograph (#1836 )
* dgl.batch
* unbatch
* fix to device
* reduce readout; segment reduce
* change batch_num_nodes|edges to function
* reduce readout/ softmax
* broadcast
* topk
* fix
* fix tf and mx
* fix some ci
* fix batch but unbatch differently
* new checkk
* upd
* upd
* upd
* idtype behavior; code reorg
* idtype behavior; code reorg
* wip: test_basics
* pass test_basics
* WIP: from nx/ to nx
* missing files
* upd
* pass test_basics:test_nx_conversion
* Fix test
* Fix inplace update
* WIP: fixing tests
* upd
* pass test_transform cpu
* pass gpu test_transform
* pass test_batched_graph
* GPU graph auto cast to int32
* missing file
* stash
* WIP: rgcn-hetero
* Fix two datasety
* upd
* weird
* Fix capsuley
* fuck you
* fuck matthias
* Fix dgmg
* fix bug in block degrees; pass rgcn-hetero
* rgcn
* gat and diffpool fix
also fix ppi and tu dataset
* Tree LSTM
* pointcloud
* rrn; wip: sgc
* resolve conflicts
* upd
* sgc and reddit dataset
* upd
* Fix deepwalk, gindt and gcn
* fix datasets and sign
* optimization
* optimization
* upd
* upd
* Fix GIN
* fix bug in add_nodes add_edges; tagcn
* adaptive sampling and gcmc
* upd
* upd
* fix geometric
* fix
* metapath2vec
* fix agnn
* fix pickling problem of block
* fix utests
* miss file
* linegraph
* upd
* upd
* upd
* graphsage
* stgcn_wave
* fix hgt
* on unittests
* Fix transformer
* Fix HAN
* passed pytorch unittests
* lint
* fix
* Fix cluster gcn
* cluster-gcn is ready
* on fixing block related codes
* 2nd order derivative
* Revert "2nd order derivative"
This reverts commit 523bf6c249bee61b51b1ad1babf42aad4167f206.
* passed torch utests again
* fix all mxnet unittests
* delete some useless tests
* pass all tf cpu tests
* disable
* disable distributed unittest
* fix
* fix
* lint
* fix
* fix
* fix script
* fix tutorial
* fix apply edges bug
* fix 2 basics
* fix tutorial
Co-authored-by: yzh119 <expye@outlook.com >
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com >
Co-authored-by: Ubuntu <ubuntu@ip-172-31-51-214.ec2.internal >
Co-authored-by: Ubuntu <ubuntu@ip-172-31-7-42.us-west-2.compute.internal >
Co-authored-by: Ubuntu <ubuntu@ip-172-31-1-5.us-west-2.compute.internal >
Co-authored-by: Ubuntu <ubuntu@ip-172-31-68-185.ec2.internal >
2020-07-28 14:30:41 +08:00
Quan (Andy) Gan
ec2e24be6d
[Feature] Slice dstnode features in NN modules ( #1838 )
...
* slice dstdata from srcdata within nn module
* a bunch of fixes
* add comment
* fix gcmc layer
* repr for blocks
* fix
* fix context
* fix
* do not copy internal columns
* docstring
2020-07-25 21:57:49 +08:00
Jinjing Zhou
33abd2757b
[HeteroGraph] Metis partition for hetero ( #1843 )
...
* metis for hetero
* fix partition
2020-07-22 16:31:10 +08:00
Da Zheng
ea420c0abd
[Distributed] Reduce memory consumption in graph partitioning ( #1823 )
...
* save mem.
* save mem.
* reduce mem
* fix test
* fix lint
* fix test
* fix.
* fix.
* fix.
* fix.
* fix lint.
* fix backend operator.
* fix tensorflow operators.
* fix.
* revert change in mxnet operator.
Co-authored-by: Ubuntu <ubuntu@ip-172-31-19-1.us-west-2.compute.internal >
2020-07-18 06:53:48 -07:00
xiang song(charlie.song)
0e92dad69a
[Transfom] Add support for to_bidirected of heterogeneous graphs ( #1793 )
...
* code
* joint_union
* joint union pass all test
* test case for test_transform
* upd
* lint
* lint
* Fix
* list
* Fix
* Fix
* update to_bidirected impl
* Fix
* Fix
* upd
* remove joint_union
Co-authored-by: Ubuntu <ubuntu@ip-172-31-51-214.ec2.internal >
2020-07-15 18:01:02 +08:00
xiang song(charlie.song)
a5722d02a4
[Transform] New ToSimple API ( #1789 )
...
* To simple
* lint
* upd
* more test
* fix
* Fix
* Fix
* add share edata
* doc
* Fix
* Fix
* Fix
* Fix
* lint
* lint
* Fix
* Fix
Co-authored-by: Ubuntu <ubuntu@ip-172-31-51-214.ec2.internal >
Co-authored-by: Mufei Li <mufeili1996@gmail.com >
2020-07-15 13:45:32 +08:00
xiang song(charlie.song)
200340ab38
[Transform] Reverse for heterogenous graph ( #1784 )
...
* reverse
* Add more test
* Fix lint
* Fix
* move to transform
* upd
* upd
* upd
* Add more test
* lint
* Fix
* Fix doc
* Fix
* test
* doc
* Fix
Co-authored-by: Ubuntu <ubuntu@ip-172-31-51-214.ec2.internal >
2020-07-13 18:05:44 +08:00
xiang song(charlie.song)
33330fcb2d
[Kernel] Line graph with new Kernel ( #1744 )
...
* line graph
* update docstring
* Add test and compile OK
* Add python API
* Fix linty
* Fix
* Fix
* upd
* upd
* Fix
* merge
* Fix
* fix
* upd
* lint
* Fix
Co-authored-by: Ubuntu <ubuntu@ip-172-31-51-214.ec2.internal >
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
2020-07-10 12:24:01 +08:00
xiang song(charlie.song)
29e6c93fba
[Kernel] Matrix toSimple ( #1756 )
...
* Matrix to simple
* Pass test
* new impl
* Fix test
* Fix lint
* trigger
* upd
* Fix comments
Co-authored-by: Ubuntu <ubuntu@ip-172-31-51-214.ec2.internal >
2020-07-09 13:50:41 +08:00
Minjie Wang
870da747ea
[CUDA][Kernel] More CUDA kernels; Standardize the behavior for sorted COO/CSR ( #1704 )
...
* add cub; array cumsum
* CSRSliceRows
* fix warning
* operator << for ndarray; CSRSliceRows
* add CSRIsSorted
* add csr_sort
* inplace coosort and outplace csrsort
* WIP: coo is sorted
* mv cuda_utils
* add AllTrue utility
* csr sort
* coo sort
* coo2csr for sorted coo arrays
* CSRToCOO from sorted
* pass tests for the new kernel changes
* cannot use inplace sort
* lint
* try fix msvc error
* Fix g.copy_to and g.asnumbits; ToBlock no longer uses CSC
* stash
* revert some hack
* revert some changes
* address comments
* fix
* fix to_block unittest
* add todo note
2020-06-28 18:37:28 +08:00
Da Zheng
04d4680d5b
[Distributed] add constraints for metis partitioning. ( #1708 )
...
* add constraint.
* fix bugs.
* temp fix.
* fix lint
* add balance_ntypes and balance_edges
* add comments.
* fix.
* fix
2020-06-27 21:27:18 -07:00
Da Zheng
4b96addca0
[BUGFIX] Fix bugs in range partitioning. ( #1703 )
...
* Fix bugs in partitioning.
* add comment.
* fix lint.
* fix test.
* add more tests.
* simplify the fix
* fix.
2020-06-26 15:59:45 -07:00
Da Zheng
41349dcef5
[Feature] Range partition ( #1522 )
...
* add reorder immutable graph.
* add python API.
* add reorder for csr.
* remove gk version.
* fix
* add cpp test.
* bug fixes
* fix tests.
* fix bugs and add check
* fix test.
* add omp.
* add comments.
* add coo reorder.
* fix a bug.
* handle reorder for different graph structues.
* fix lint.
* fix.
* add original ids.
* reshuffle nodes before metis partition.
* inner nodes are in contiguous Id range.
* reshuffle nodes/edges when partitioning.
* load partition return graph partition book.
* use inner_node/inner_edges
* add and test range partition book.
* count inner_edge correctly.
* fix lint.
* fix lint.
* fix lint.
* fix errors.
* fix errors.
* fix for TF.
* fix.
* fix.
* change docstring.
* support logical and.
* add comments.
* avoid copy.
* fix
* update docstring.
* fix a bug.
* add range search.
* fix
* fix a bug.
* add more tests.
* load graph partition book.
* support shared memory for range partition book.
* fix a bug.
* fix.
* fix lint.
* remove check
* fix test.
* remove num_nodes and num_edges
* fix lint.
* fix graph partition book.
* address comments.
* use makedirs.
* fix compile
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com >
Co-authored-by: Chao Ma <mctt90@gmail.com >
2020-06-15 11:43:04 -07:00
Jinjing Zhou
dc8ca88e58
[Refactor] Explicit dtype for HeteroGraph ( #1467 )
...
* 111
* 111
* lint
* lint
* lint
* lint
* fix
* lint
* try
* fix
* lint
* lint
* test
* fix
* ttt
* test
* fix
* fix
* fix
* mxnet
* 111
* fix 64bits computation
* pylint
* roll back
* fix
* lint
* fix hetero_from_relations
* remove index_dtype in to_homo and to_hetero
* fix
* fix
* fix
* fix
* remove default
* fix
* lint
* fix
* fix error message
* fix error
* lint
* macro dispatch
* try
* lint
* remove nbits
* error message
* fix
* fix
* lint
* lint
* lint
* fix
* lint
* fix
* fix random walk
* lint
* lint
* fix
* fix
* fix
* lint
* fix
* lint
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
2020-05-08 14:26:28 +08:00
Mufei Li
bd1e48a51e
[Bug] Fix khop graph ( #1433 )
...
* Update
* Update
* Update
2020-04-10 02:18:45 +08:00
Quan (Andy) Gan
e3a9a6bba8
add an optional include_dst_in_src argument ( #1401 )
2020-03-30 12:06:12 +08:00
xiang song(charlie.song)
67cb7a43a0
[Feature] Deprecate multigraph ( #1389 )
...
* Deprecate multi-graph
* Handle heterograph and edge_ids
* lint
* Fix
* Remove multigraph in C++ end
* Fix lint
* Add some test and fix something
* Fix
* Fix
* upd
* Fix some test case
* Fix
* Fix
Co-authored-by: Ubuntu <ubuntu@ip-172-31-51-214.ec2.internal >
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com >
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
2020-03-28 21:31:24 +08:00
Quan (Andy) Gan
bbfff8ce76
[Feature] Casting between DGLGraph and DGLHeteroGraph ( #1391 )
...
* [Feature] Casting between DGLGraph and DGLHeteroGraph
* lint
* address comments
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com >
2020-03-27 13:34:55 +08:00
Quan (Andy) Gan
b9c65e91ef
[BUG] Another fix on remove edges when all edges are removed ( #1386 )
...
* [BUG] Another fix on remove edges when all edges are removed
* fix mxnet
2020-03-23 19:04:52 +08:00
Quan (Andy) Gan
d27b485912
[BUG] Fix remove_edges crashing with empty edge ID tensors ( #1384 )
2020-03-22 21:09:44 +08:00
Da Zheng
10253a5c07
[BUGFIX] Fix is_multigraph in the construction from scipy coo matrix ( #1357 )
...
* fix is_multigraph in from_coo.
* add tests for partition.
* fix.
* Revert "add tests for partition."
This reverts commit cb8c8555da3e0c70a482c2d639adce2943475bfc.
* fix everywhere from_scipy_sparse_matrix is used.
2020-03-22 00:30:27 -07:00
Mufei Li
0f40c6e49c
[Hetero] Replace card with num_nodes
...
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
2020-03-20 16:56:53 +08:00
Minjie Wang
6111ea4665
[Model] RGCN mini-batch training; Better bipartite graph support ( #1337 )
...
* change the model to use node embedding only
* minibatch training
* add readme
* small fix
* backward-compatible
* backward compatible
* modify to_block; rgcn changed
* fix
* fix transform
* fix bug in unittest script
* docstring
* fix lint
* add tests
* address comments; fix offline eval
* gitignore
2020-03-16 18:58:06 +08:00
Da Zheng
1a584ced11
[Feature] Improve the speed of to_bidirected ( #1327 )
...
* make graph symmetric
* call gklib routine.
* use gklib version except in windows.
* minor errors.
* fix test.
* update doc.
* fix a compile error.
* fix.
* add comments for the new C API.
* fix a bug.
* address comments.
* fix compile error.
* fix comment.
2020-03-10 22:44:32 -07:00
xiang song(charlie.song)
7b3a7b1438
[Optimization] Optimize CompactGraph ( #1328 )
...
* compact
* prealloc memory for hashtable
* Fix
* upd
* Reduce memory
* upd
Co-authored-by: Ubuntu <ubuntu@ip-172-31-51-214.ec2.internal >
2020-03-10 12:54:12 +08:00
Da Zheng
0e153c4baf
[Feature] add metis partitioning to DGL ( #1308 )
...
* add metis.
* add test.
* construct partition id.
* link to METIS github repo.
* update metis.
* add a tool for partitioning a graph.
* update metis.
* update.
* update.
* fix metis.
* fix lint
* fix indent.
* another way of building metis.
* disable metis in windows.
* test windows
* fix.
* disable metis for windows properly.
* fix for tensorflow.
* skip test for gpu.
* make graph symmetric
* address comments.
* more comments.
* fix compile
* fix a bug.
* add test.
* change the default #hops of HALO nodes.
Co-authored-by: Ubuntu <ubuntu@ip-172-31-26-167.us-east-2.compute.internal >
2020-03-07 23:24:03 -08:00
Quan (Andy) Gan
a9520f71ce
[Model][Sampler] GraphSAGE model, bipartite graph conversion & remove edges API ( #1297 )
...
* remove edge and to bipartite and graphsage with sampling
* fixes
* fixes
* fixes
* reenable multigpu training
* fixes
* compatibility in DGLGraph
* rename to compact_as_bipartite
* bugfix
* lint
* add offline inference
* skip GPU tests
* fix
* addresses comments
* fix
* fix
* fix
* more tests
* more docs and unit tests
* workaround for empty slice on empty data
2020-03-08 02:38:29 +08:00
Zihao Ye
09ade2f265
upd ( #1321 )
2020-03-06 14:17:48 +08:00
Quan (Andy) Gan
c3a33407fe
[Feature] PinSAGE sampler ( #1249 )
...
* [WIP] PinSAGE operators
* moved the edge remapping mess into C
* some docstrings
* lint
* lint x2
* lint x3
* skip gpu test on topk
* extend pinsage to any metapath
* lint x4
* addresses #1265
* add always_preserve (fixes #1266 ) and fix a silly bug
* disable gpu test on compaction
* lint
* fix a horrible bug and add more tests
* lint
* addresses comments
* lint
* bugfix
* addresses comments
Co-authored-by: Minjie Wang <minjie.wang@nyu.edu >
2020-02-28 20:46:13 +08:00
Minjie Wang
f8d4264e5b
[Feature] Neighbor-hood based sampling APIs ( #1251 )
...
* WIP: working on random choices
* light slice
* basic CPU impl
* add python binding; fix CreateFromCOO and CreateFromCSR returning unitgraph
* simple test case works
* fix bug in slicing probability array
* fix bug in getting the correct relation graph
* fix bug in creating placeholder graph
* enable omp
* add cpp test
* sample topk
* add in|out_subgraph
* try fix lint; passed all unittests
* fix lint
* fix msvc compile; add sorted flag and constructors
* fix msvc
* coosort
* COOSort; CSRRowWiseSampling; CSRRowWiseTopk
* WIP: remove DType in CSR and COO; Restrict data array to be IdArray
* fix all CSR ops for missing data array
* compiled
* passed tests
* lint
* test sampling out edge
* test different per-relation fanout/k values
* fix bug in random choice
* finished cpptest
* fix compile
* Add induced edges
* add check
* fixed bug in sampling on hypersparse graph; add tests
* add ascending flag
* in|out_subgraph returns subgraph and induced eid
* address comments
* lint
* fix
2020-02-17 21:25:21 +08:00
Da Zheng
e890a899fc
Partition a graph with HALO nodes ( #1076 )
...
* get subgraph with halo.
* add partition function.
* add comment.
* parallel partition.
* fix a compilation error.
* fix lint error.
* address comments.
* add comments.
* fix for TF.
2019-12-23 08:46:53 -08:00
VoVAllen
e16667bfb9
[Fix] Fix Transform ( #876 )
...
* add transform
* lint
* lint
* fix
* fixmx
* fix
* add test
* fix typo
* fix default num_classes
* change to non-inplace operation
* fix lint
* fix
* fix
* fix lint
* fixlint
2019-09-26 21:21:53 -04:00
VoVAllen
bf8bb58f60
[Transform] Add to_self_loop, remove_self_loop, onehot_degree transform ( #862 )
...
* add transform
* lint
* lint
* fix
* fixmx
* fix
* add test
* fix typo
* fix default num_classes
* change to non-inplace operation
* fix lint
* fix
2019-09-21 00:05:38 +08:00