1. 07 Nov, 2022 3 commits
  2. 06 Nov, 2022 1 commit
  3. 19 Sep, 2022 1 commit
    • Xin Yao's avatar
      [Feature] Bump DLPack to v0.7 and decouple DLPack from the core library (#4454) · cded5b80
      Xin Yao authored
      * rename `DLContext` to `DGLContext`
      
      * rename `kDLGPU` to `kDLCUDA`
      
      * replace DLTensor with DGLArray
      
      * fix linting
      
      * Unify DGLType and DLDataType to DGLDataType
      
      * Fix FFI
      
      * rename DLDeviceType to DGLDeviceType
      
      * decouple dlpack from the core library
      
      * fix bug
      
      * fix lint
      
      * fix merge
      
      * fix build
      
      * address comments
      
      * rename dl_converter to dlpack_convert
      
      * remove redundant comments
      cded5b80
  4. 15 Sep, 2022 1 commit
    • Xin Yao's avatar
      [Feature] Import PyTorch's CUDA stream management (#4503) · 9a00cf19
      Xin Yao authored
      * add set_stream
      
      * add .record_stream for NDArray and HeteroGraph
      
      * refactor dgl stream Python APIs
      
      * test record_stream
      
      * add unit test for record stream
      
      * use pytorch's stream
      
      * fix lint
      
      * fix cpu build
      
      * address comments
      
      * address comments
      
      * add record stream tests for dgl.graph
      
      * record frames and update dataloder
      
      * add docstring
      
      * update frame
      
      * add backend check for record_stream
      
      * remove CUDAThreadEntry::stream
      
      * record stream for newly created formats
      
      * fix bug
      
      * fix cpp test
      
      * fix None c_void_p to c_handle
      9a00cf19
  5. 06 Sep, 2022 1 commit
    • Chang Liu's avatar
      [Feature] Unify the cuda stream used in core library (#4480) · 1c9d2a03
      Chang Liu authored
      
      
      * Use an internal cuda stream for CopyDataFromTo
      
      * small fix white space
      
      * Fix to compile
      
      * Make stream optional in copydata for compile
      
      * fix lint issue
      
      * Update cub functions to use internal stream
      
      * Lint check
      
      * Update CopyTo/CopyFrom/CopyFromTo to use internal stream
      
      * Address comments
      
      * Fix backward CUDA stream
      
      * Avoid overloading CopyFromTo()
      
      * Minor comment update
      
      * Overload copydatafromto in cuda device api
      Co-authored-by: default avatarxiny <xiny@nvidia.com>
      1c9d2a03
  6. 09 Apr, 2022 1 commit
  7. 24 Mar, 2022 1 commit
  8. 21 Feb, 2022 1 commit
    • Quan (Andy) Gan's avatar
      [Bugfix] Bug fixes in new dataloader (#3727) · 3f138eba
      Quan (Andy) Gan authored
      
      
      * fixes
      
      * fix
      
      * more fixes
      
      * update
      
      * oops
      
      * lint?
      
      * temporarily revert - will fix in another PR
      
      * more fixes
      
      * skipping mxnet test
      
      * address comments
      
      * fix DDP
      
      * fix edge dataloader exclusion problems
      
      * stupid bug
      
      * fix
      
      * use_uvm option
      
      * fix
      
      * fixes
      
      * fixes
      
      * fixes
      
      * fixes
      
      * add evaluation for cluster gcn and ddp
      
      * stupid bug again
      
      * fixes
      
      * move sanity checks to only support DGLGraphs
      
      * pytorch lightning compatibility fixes
      
      * remove
      
      * poke
      
      * more fixes
      
      * fix
      
      * fix
      
      * disable test
      
      * docstrings
      
      * why is it getting a memory leak?
      
      * fix
      
      * update
      
      * updates and temporarily disable forkingpickler
      
      * update
      
      * fix?
      
      * fix?
      
      * oops
      
      * oops
      
      * fix
      
      * lint
      
      * huh
      
      * uh
      
      * update
      
      * fix
      
      * made it memory efficient
      
      * refine exclude interface
      
      * fix tutorial
      
      * fix tutorial
      
      * fix graph duplication in CPU dataloader workers
      
      * lint
      
      * lint
      
      * Revert "lint"
      
      This reverts commit 805484dd553695111b5fb37f2125214a6b7276e9.
      
      * Revert "lint"
      
      This reverts commit 0bce411b2b415c2ab770343949404498436dc8b2.
      
      * Revert "fix graph duplication in CPU dataloader workers"
      
      This reverts commit 9e3a8cf34c175d3093c773f6bb023b155f2bd27f.
      Co-authored-by: default avatarxiny <xiny@nvidia.com>
      Co-authored-by: default avatarJinjing Zhou <VoVAllen@users.noreply.github.com>
      3f138eba
  9. 21 Jan, 2022 1 commit
    • Xin Yao's avatar
      [Feature] Pin dgl.graph to the page-locked memory (#3616) · 40b44a43
      Xin Yao authored
      
      
      * implement pin_memory/unpin_memory/is_pinned for dgl.graph
      
      * update python docstring
      
      * update c++ docstring
      
      * add test
      
      * fix the broken UnifiedTensor
      
      * eliminate extra context parameter for pin/unpin
      
      * fix linting
      
      * fix typo
      
      * disable new format materialization for pinned graphs
      
      * update python doc for pin_memory_
      
      * fix unit test
      
      * update doc
      
      * change unitgraph and heterograph's PinMemory to in-place
      
      * update comments for NDArray's PinMemory_ and PinData
      
      * update doc
      Co-authored-by: default avatarJinjing Zhou <VoVAllen@users.noreply.github.com>
      40b44a43
  10. 01 Sep, 2021 1 commit
  11. 30 Jul, 2020 1 commit
  12. 28 Jul, 2020 2 commits
    • Qidong Su's avatar
      [Feature] Shared memory utilities (#1807) · 40950629
      Qidong Su authored
      
      
      * update
      
      * update
      
      * update
      
      * update
      
      * fix
      
      * update
      
      * fix
      
      * update
      
      * update
      
      * win32
      
      * update
      
      * fix
      
      * update
      
      * update
      
      * update
      
      * updat
      
      * update
      
      * update
      
      * fix
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * fix
      
      * TODO
      
      * 111
      
      * fix
      
      * minor fix
      
      * minor fix
      
      * fox
      
      * Update shared_mem_manager.cc
      
      * update
      
      * update
      
      * update
      
      * update metis
      
      * update metis
      
      * update
      Co-authored-by: default avatarVoVAllen <jz1749@nyu.edu>
      Co-authored-by: default avatarJinjing Zhou <VoVAllen@users.noreply.github.com>
      40950629
    • 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
  13. 10 Jul, 2020 1 commit
  14. 04 Jul, 2020 1 commit
  15. 14 May, 2020 1 commit
  16. 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
  17. 28 Mar, 2020 1 commit
  18. 27 Mar, 2020 1 commit
  19. 07 Mar, 2020 1 commit
  20. 06 Mar, 2020 1 commit
  21. 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
  22. 10 Feb, 2020 1 commit
  23. 04 Feb, 2020 1 commit
    • Quan (Andy) Gan's avatar
      [Feature] Preference to COO for "hypersparse" unit graphs & graph compaction (#1238) · eeeb52f4
      Quan (Andy) Gan authored
      * unit graph that prefers coo queries
      
      * auto detect coo preference
      
      * forgot some functions
      
      * disable lint on detect_prefer_coo
      
      * reorg
      
      * change comment
      
      * lint
      
      * fix
      
      * move array_utils.h to src
      
      * compact graph impl
      
      * fix redundant copying in idhashmap
      
      * docstring
      
      * moving preference detection to C
      
      * lint
      
      * fix unit test & address comments
      
      * hypersparse autorestrict
      
      * docstring & fix
      
      * revert copyto and asnumbits
      
      * fix stupid bug
      
      * lint
      
      * leave a TODO for sorted COO
      
      * fixing same node type mapping to different id in different graphs
      
      * addresses comments
      
      * made induced nodes a feautre column
      
      * lint?
      eeeb52f4
  24. 17 Sep, 2019 1 commit
    • Minjie Wang's avatar
      [Hetero] New syntax (#824) · 9b4d6079
      Minjie Wang authored
      * WIP. remove graph arg in NodeBatch and EdgeBatch
      
      * refactor: use graph adapter for scheduler
      
      * WIP: recv
      
      * draft impl
      
      * stuck at bipartite
      
      * bipartite->unitgraph; support dsttype == srctype
      
      * pass test_query
      
      * pass test_query
      
      * pass test_view
      
      * test apply
      
      * pass udf message passing tests
      
      * pass quan's test using builtins
      
      * WIP: wildcard slicing
      
      * new construct methods
      
      * broken
      
      * good
      
      * add stack cross reducer
      
      * fix bug; fix mx
      
      * fix bug in csrmm2 when the CSR is not square
      
      * lint
      
      * removed FlattenedHeteroGraph class
      
      * WIP
      
      * prop nodes, prop edges, filter nodes/edges
      
      * add DGLGraph tests to heterograph. Fix several bugs
      
      * finish nx<->hetero graph conversion
      
      * create bipartite from nx
      
      * more spec on hetero/homo conversion
      
      * silly fixes
      
      * check node and edge types
      
      * repr
      
      * to api
      
      * adj APIs
      
      * inc
      
      * fix some lints and bugs
      
      * fix some lints
      
      * hetero/homo conversion
      
      * fix flatten test
      
      * more spec in hetero_from_homo and test
      
      * flatten using concat names
      
      * WIP: creators
      
      * rewrite hetero_from_homo in a more efficient way
      
      * remove useless variables
      
      * fix lint
      
      * subgraphs and typed subgraphs
      
      * lint & removed heterosubgraph class
      
      * lint x2
      
      * disable heterograph mutation test
      
      * docstring update
      
      * add edge id for nx graph test
      
      * fix mx unittests
      
      * fix bug
      
      * try fix
      
      * fix unittest when cross_reducer is stack
      
      * fix ci
      
      * fix nx bipartite bug; docstring
      
      * fix scipy creation bug
      
      * lint
      
      * fix bug when converting heterograph from homograph
      
      * fix bug in hetero_from_homo about ntype order
      
      * trailing white
      
      * docstring fixes for add_foo and data views
      
      * docstring for relation slice
      
      * to_hetero and to_homo with feature support
      
      * lint
      
      * lint
      
      * DGLGraph compatibility
      
      * incidence matrix & docstring fixes
      
      * example string fixes
      
      * feature in hetero_from_relations
      
      * deduplication of edge types in to_hetero
      
      * fix lint
      
      * fix
      9b4d6079
  25. 27 Jul, 2019 1 commit
    • Minjie Wang's avatar
      [Hetero] Heterograph C++ implementation; Bipartite and Python wrapper (#725) · 7ad663c3
      Minjie Wang authored
      * finish bipartite graph implementation; compiled
      
      * finished heterograph implementation; compiled
      
      * WIP: apis
      
      * C API codes
      
      * compiled
      
      * WIP: python
      
      * HeteroGraphIndex
      
      * WIP: test
      
      * add DGLContext support in ffi
      
      * fix bug in has edge
      
      * unittests except edge subgraph
      
      * edge subgraph
      
      * fix lint
      
      * address comments
      
      * poke ci
      
      * try fix
      
      * fix msvc
      7ad663c3