949f87cc1d
[FFI] Change enumeration name in DGLObjectTypeCode to avoid conflict with Pytorch ( #4905 )
2022-11-16 15:25:48 +08:00
cded5b80fe
[Feature] Bump DLPack to v0.7 and decouple DLPack from the core library ( #4454 )
...
* 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
2022-09-19 16:02:43 +08:00
9a00cf194f
[Feature] Import PyTorch's CUDA stream management ( #4503 )
...
* 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
2022-09-15 16:54:52 +08:00
1c9d2a0302
[Feature] Unify the cuda stream used in core library ( #4480 )
...
* 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: xiny <xiny@nvidia.com >
2022-09-06 10:14:55 +08:00
3c129ad71f
[Bugfix] Cython CAPI holding GIL causes deadlock when Python callback is asynchronous ( #4036 )
...
* cython nogil
* move APIs to internal and add unit test
* fix lint
* disable callback array test
2022-05-25 10:02:44 +08:00
e9440acb06
[TF] TF backend fix and new logic to choose backend ( #1393 )
...
* TF backend fix and new logic to choose backend
* fix
* fix
* fix
* fix
* fix backend
* fix
* dlpack alignment
* add flag
* flag
* lint
* lint
* remove unused
* several fixes
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
2020-03-30 18:45:17 +08:00
90e78c5854
[FFI] FFI container support, custom structure extension via Object ( #693 )
...
* WIP: import tvm runtime node system
* WIP: object system
* containers
* tested basic container composition
* tested custom object
* fix setattr bug
* tested object container return
* fix lint
* some comments about get/set state
* fix lint
* fix lint
* update cython
* fix cython
* ffi doc
* fix doc
2019-07-01 14:40:07 -04:00
fb6af16f83
[BUG] Fix crash during Cython build ( #500 )
2019-04-17 14:18:22 +08:00
bfdd1eaa44
[Feature] Create shared memory graph store. ( #468 )
...
* accelerate gcn_ns.
* add timing.
* run infer with whole graph.
* distributed gcn_ns.
* reconstruct gcn_ns.
* minor fix.
* change graphsage_cv for numa.
* fix #OMP threads.
* accelerate graphsage_cv.
* fix a weird bug.
* add profiler in graphsage_cv.
* accelerate graphsage_cv.
manually aggregate neighbors' embeddings with pull.
* load csr directly in gcn_ns_sc.
* parallel sort for graph index.
* Revert "parallel sort for graph index."
This reverts commit 86fe2c7117fe5e56b0d481b39849c258b166945b.
* run gcn_ns_sc on GPUs.
* acc gcn_cv_sc.
* change gcn_cv for numa.
* fix gcn_cv to use numa and gpu.
* improve graphsage_cv to use numa and gpu.
* improve gcn_ns.
* improve graphsage_cv.
* init shared memory graph store.
* fix.
* enable init ndata.
* improve tests.
* add bidirectional communication.
* link to rt.
* fix compilation error.
* fix shared memory init.
* use MessageQueue for inter-process communication.
* reconstruct immutable graph csr.
* fix gcn.
* load csr to shared memory.
* fix minor bugs.
* add comments.
* refactor SharedMemory.
* fix bugs in ImmutableGraph.
* create CSR graph from shared memory.
* add more test for loading a csr graph.
* terminate graph store properly.
* allow initializing ndata in the graph store server.
* use RPC for inter-process communication.
* a script for loading a graph.
* allow customizing port.
* list all ndata and edata.
* support dtype.
* reorganize SharedMemoryGraphStore.
* fix ndata shape.
* reconstruct gcn_ns.
* print info.
* set omp in gcn_ns.
* reset sampling examples.
* fix lint.
* fix lint.
* reset gcn.
* disable shared memory in windows.
* fix.
* fix.
* reset changes.
* revert nodeflow changes.
* fix cmake.
* fix test.
* fix test.
* fix test.
* fix test.
* add comments.
* fix test.
* move vector out.
* fix lint.
* fix lint.
* move SharedMemory.
* update cmake.
* update comment.
* fix comments.
* Revert "update cmake."
This reverts commit 592445e37077f70a6e3f2e5245f9a3d086b04f3b.
* update cmake.
* add comments.
* rename.
* change the comment.
* fix a bug.
* rename.
* add comments.
* add comments.
* add init_edata.
* rewrite memory alloc.
* move vector to CSR.
* fix.
* init data.
* Revert "init data."
This reverts commit 2b217b9553911b7dd84a9f1d9b68430b5aa18e23.
* init data.
* init new columns correctly.
2019-04-08 12:09:02 -07:00
ca2a7e1ca1
[Refactor] Nodeflow, sampling, CAPI ( #430 )
...
* enable cython
* add helper function and data structure for void_p vector return
* move sampler from graph index to contrib.sampling
* WIP
* WIP
* refactor layer sampling
* pass tests
* fix lint
* fix graphsage
* remove comments
* pickle test
* fix comments
* update dev guide for cython build
2019-03-05 14:07:22 -05:00
a1d50f0f53
[Refactor] Rename before release ( #261 )
...
* include/dgl/runtime
* include
* src/runtime
* src/graph
* src/scheduler
* src
* clean up CMakeLists
* further clean up in cmake
* install commands
* python/dgl/_ffi/_cython
* python/dgl/_ffi/_ctypes
* python/dgl/_ffi
* python/dgl
* some fix
* copy right
2018-12-05 16:45:36 -05:00
2694b12725
import ffi solution from TVM
2018-09-05 10:51:31 -04:00
czkkkkkk