1. 28 Jun, 2020 1 commit
    • Minjie Wang's avatar
      [CUDA][Kernel] More CUDA kernels; Standardize the behavior for sorted COO/CSR (#1704) · 870da747
      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
      870da747
  2. 19 Jun, 2020 1 commit
    • Minjie Wang's avatar
      [CUDA] Many CUDA operators; Prepare for DGLGraph on CUDA (#1660) · f1b19a6b
      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
      f1b19a6b
  3. 15 Jun, 2020 2 commits
    • Da Zheng's avatar
      [Feature] Range partition (#1522) · 41349dce
      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: default avatarxiang song(charlie.song) <classicxsong@gmail.com>
      Co-authored-by: default avatarChao Ma <mctt90@gmail.com>
      41349dce
    • Minjie Wang's avatar
      [Kernel] CUDA CSR2COO COOSort COO2CSR (#1620) · d6d517bb
      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: default avatarJinjing Zhou <VoVAllen@users.noreply.github.com>
      Co-authored-by: default avatarChao Ma <mctt90@gmail.com>
      Co-authored-by: default avatarxiang song(charlie.song) <classicxsong@gmail.com>
      d6d517bb
  4. 13 Mar, 2020 1 commit
  5. 02 Mar, 2020 1 commit
    • Minjie Wang's avatar
      [Feature] Improve sampling speed; Better pickle/unpickle; other fixes (#1299) · 5dd35580
      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
      5dd35580
  6. 28 Sep, 2019 1 commit
    • Da Zheng's avatar
      Sort csr (#886) · 65e1ba4f
      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.
      65e1ba4f
  7. 17 Jul, 2019 1 commit
    • Minjie Wang's avatar
      [Refactor] Separating graph and sparse matrix operations (#699) · b0d9e7aa
      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
      b0d9e7aa