- 28 Jun, 2020 1 commit
-
-
Minjie Wang authored
* 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
-
- 19 Jun, 2020 1 commit
-
-
Minjie Wang authored
* add cuda utils; change g.to; add g.device * split array.h into several headers * cuda index select * file * three cuda kernels * add cuda elementwise arith and several others * cuda CSRIsNonZero * fix lint * lint * lint * fix bug in changing ctx to property * address comments * remove unused codes * address comments
-
- 15 Jun, 2020 2 commits
-
-
Da Zheng authored
* 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>
-
Minjie Wang authored
* add cuda source * moving codes from kernel2 branch * operator overloading * Better error message for unsupported device * fix c tests * coo sort using cusparse * move test_rpc to distributed * lint * address comments and add utests Co-authored-by:
Jinjing Zhou <VoVAllen@users.noreply.github.com> Co-authored-by:
Chao Ma <mctt90@gmail.com> Co-authored-by:
xiang song(charlie.song) <classicxsong@gmail.com>
-
- 13 Mar, 2020 1 commit
-
-
xiang song(charlie.song) authored
* Fix coo_sort * Add COOSort test * Fix * omp * coo2csr can benefit from sorted coo * Add more test code * Add LIBCXX_ENABLE_PARALLEL_ALGORITHMS to Makefile * upd * Turn off LIBCXX_ENABLE_PARALLEL_ALGORITHMS by default Co-authored-by:Ubuntu <ubuntu@ip-172-31-51-214.ec2.internal>
-
- 02 Mar, 2020 1 commit
-
-
Minjie Wang authored
* improve performance of sample_neighbors * some more improve * test script * benchmarks * multi process * update more tests * WIP * adding two API for state saving * add create from state * upd test * missing file * wip: pickle/unpickle * more c apis * find the problem of empty data array * add null array; pickling speed is bad * still bad perf * still bad perf * wip * fix the pickle speed test; now everything looks good * minor fix * bugfix * some lint fix * address comments * more fix * fix lint * add utest for random.choice * add utest for dgl.rand_graph * fix cpp utests * try fix ci * fix bug in TF backend * upd choice docstring * address comments * upd * try fix compile * add comment
-
- 28 Sep, 2019 1 commit
-
-
Da Zheng authored
* sort * sort in parallel. * fix a bug in sorting adj * rename. * add more comments. * accelerate GetData * fix tests. * avoid sorting multiple times. * add test. * change back. * sort. * add sort_csr. * Fix a bug. * fix. * revert modifcation. * rename * speed up EdgeIds.
-
- 17 Jul, 2019 1 commit
-
-
Minjie Wang authored
* WIP: array refactoring * WIP: implementation * wip * most csr part * WIP: on coo * WIP: coo * finish refactoring immutable graph * compiled * fix undefined ndarray copy bug; add COOToCSR when coo has no data array * fix bug in COOToCSR * fix bug in CSR constructor * fix bug in in_edges(vid) * fix OutEdges bug * pass test_graph * pass test_graph * fix bug in CSR constructor * fix bug in CSR constructor * fix bug in CSR constructor * fix stupid bug * pass gpu test * remove debug printout * fix lint * rm biparate grpah * fix lint * address comments * fix bug in Clone * cpp utests
-