1. 25 Dec, 2020 1 commit
    • Quan (Andy) Gan's avatar
      [Performance] Use allocator from PyTorch if possible (#2328) · 9a7235fa
      Quan (Andy) Gan authored
      * first commit
      
      * some thoughts
      
      * move around
      
      * more commit
      
      * more fixes
      
      * now it uses torch allocator
      
      * fix symbol export error
      
      * fix
      
      * fixes
      
      * test fix
      
      * add script
      
      * building separate library per version
      
      * fix for vs2019
      
      * more fixes
      
      * fix on windows build
      
      * update jenkinsfile
      
      * auto copy built dlls for windows
      
      * lint and installation guide update
      
      * fix
      
      * specify conda environment
      
      * set environment for ci
      
      * fix
      
      * fix
      
      * fix
      
      * fix again
      
      * revert
      
      * fix cmake
      
      * fix
      
      * switch to using python interpreter path
      
      * remove scripts
      
      * debug
      
      * oops sorry
      
      * Update index.rst
      
      * Update index.rst
      
      * copies automatically, no need for this
      
      * do not print message if library not found
      
      * tiny fixes
      
      * debug on nightly
      
      * replace add_compile_definitions to make CMake 3.5 happy
      
      * fix linking to wrong lib for multiple pytorch envs
      
      * changed building strategy
      
      * fix nightly
      
      * fix windows
      
      * fix windows again
      
      * setup bugfix
      
      * address comments
      
      * change README
      9a7235fa
  2. 17 Dec, 2020 1 commit
  3. 10 Dec, 2020 1 commit
  4. 27 Nov, 2020 1 commit
  5. 26 Nov, 2020 1 commit
  6. 22 Nov, 2020 1 commit
  7. 17 Nov, 2020 1 commit
  8. 13 Nov, 2020 1 commit
  9. 15 Sep, 2020 1 commit
  10. 11 Sep, 2020 1 commit
  11. 14 Aug, 2020 1 commit
  12. 30 Jul, 2020 1 commit
  13. 28 Jul, 2020 1 commit
    • Minjie Wang's avatar
      [Refactor][Graph] Merge DGLGraph and DGLHeteroGraph (#1862) · 44089c8b
      Minjie Wang authored
      
      
      * 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: default avatarUbuntu <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: default avatarUbuntu <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: default avataryzh119 <expye@outlook.com>
      Co-authored-by: default avatarxiang song(charlie.song) <classicxsong@gmail.com>
      Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-51-214.ec2.internal>
      Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-7-42.us-west-2.compute.internal>
      Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-1-5.us-west-2.compute.internal>
      Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-68-185.ec2.internal>
      44089c8b
  14. 13 Jul, 2020 1 commit
    • Zihao Ye's avatar
      [Feature] Autograd of gspmm and gsddmm on PyTorch/MXNet/Tensorflow (#1680) · f25b1a06
      Zihao Ye authored
      * init
      
      * reverse(by minjie
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * gpu
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * udp
      
      * upd
      
      * upd
      
      * imidiot
      
      * fix
      
      * upd
      
      * upd
      
      * upd
      
      * udp
      
      * upd
      
      * upd
      
      * fix
      
      * udp
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * fix
      
      * remove redundency
      
      * upd
      
      * upd
      
      * upd
      
      * cache
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * udp
      
      * upd
      
      * trigger
      
      * upd
      
      * fix
      
      * upd
      
      * unused import
      
      * upd
      
      * upd
      f25b1a06
  15. 10 Jul, 2020 1 commit
  16. 09 Jul, 2020 2 commits
  17. 05 Jul, 2020 1 commit
  18. 30 Jun, 2020 1 commit
  19. 28 Jun, 2020 2 commits
    • 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
    • Quan (Andy) Gan's avatar
      [Bug] Fix #1712 (#1714) · 8b539079
      Quan (Andy) Gan authored
      * fix debug compilation
      
      * fix
      
      * revert
      8b539079
  20. 22 Jun, 2020 1 commit
    • Zihao Ye's avatar
      [kernel] New SpMM & SDDMM kernel on CPU and CUDA (#1644) · 071cba1f
      Zihao Ye authored
      * udp
      
      * simplify
      
      * sddmm dot cpu
      
      * upd
      
      * format
      
      * upd
      
      * compatible with MJ's PR
      
      * lint
      
      * upd
      
      * upd
      
      * upd
      
      * python end
      
      * upd
      
      * upd
      
      * lint
      
      * lint
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * lint
      
      * fix mxnet
      
      * upd
      
      * lint
      
      * use minjie's ptr
      
      * macro
      
      * upd
      
      * reorg
      
      * lint
      
      * fix corner cases
      
      * upd
      
      * enrich cpu docs
      
      * upd
      
      * upd
      
      * lint
      
      * lint
      
      * pylint
      
      * sx review
      
      * improve docstring
      
      * python doc
      
      * upd
      
      * restructure
      
      * lint
      
      * upd test
      
      * upd
      
      * pylint
      
      * fix corner cases and test
      071cba1f
  21. 21 Jun, 2020 1 commit
  22. 19 Jun, 2020 2 commits
  23. 15 Jun, 2020 1 commit
    • 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
  24. 11 Jun, 2020 1 commit
  25. 08 May, 2020 1 commit
    • Jinjing Zhou's avatar
      [Refactor] Explicit dtype for HeteroGraph (#1467) · dc8ca88e
      Jinjing Zhou authored
      
      
      * 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: default avatarMinjie Wang <wmjlyjemaine@gmail.com>
      dc8ca88e
  26. 28 Apr, 2020 1 commit
  27. 06 Apr, 2020 1 commit
  28. 30 Mar, 2020 1 commit
  29. 15 Mar, 2020 1 commit
  30. 13 Mar, 2020 1 commit
  31. 10 Mar, 2020 1 commit
  32. 07 Mar, 2020 3 commits
  33. 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
  34. 28 Feb, 2020 1 commit
    • Quan (Andy) Gan's avatar
      [Feature] PinSAGE sampler (#1249) · c3a33407
      Quan (Andy) Gan authored
      
      
      * [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: default avatarMinjie Wang <minjie.wang@nyu.edu>
      c3a33407
  35. 17 Feb, 2020 1 commit
    • Minjie Wang's avatar
      [Feature] Neighbor-hood based sampling APIs (#1251) · f8d4264e
      Minjie Wang authored
      * 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
      f8d4264e