文件历史

作者	SHA1	备注	提交日期
Chang Liu	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
Xin Yao	1feec870c1	[Bugfix] Add CUDA context availability check before setting curand seed (#4223 )	2022-07-09 10:19:10 +08:00
Quan (Andy) Gan	3f138ebaaf	[Bugfix] Bug fixes in new dataloader (#3727 ) * 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: xiny <xiny@nvidia.com> Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>	2022-02-22 01:30:31 +08:00
Xin Yao	40b44a43a9	[Feature] Pin dgl.graph to the page-locked memory (#3616 ) * 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: Jinjing Zhou <VoVAllen@users.noreply.github.com>	2022-01-21 16:13:12 +08:00
Rhett Ying	e234fcfa8f	[Feature] enable create/set/free cuda stream for internal use (#3334 ) * [Feature] enable create/set/free cuda stream for internal use * add unit test * fix unit test failure on mxnet and tf * refactor stream wrapper * fix lint error * fix lint error	2021-09-29 15:35:02 +08:00
David Min	905c0aa578	[Feature][Performance][GPU] Introducing UnifiedTensor for efficient zero-copy host memory access from GPU (#3086 ) * Add pytorch-direct version * Initial commit of unified tensor * Merge branch 'master' of https://github.com/davidmin7/dgl * Remove unnecessary things * Fix error message * Fix/Add descriptions * whitespace fix * add unpin * disable IndexSelectCPUFromGPU with no CUDA * add a newline for unified_tensor.py * Apply changes based on feedback * add 'os' module * skip unified tensor unit test for cpu only * Update tests/pytorch/test_unified_tensor.py Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com> * reflect feedback Co-authored-by: shhssdm <shhssdm@gmail.com> Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com> Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>	2021-07-17 00:06:20 +08:00
Lingfan Yu	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
Minjie Wang	820c6b9ed6	Add lint script and fix cpplint errors	2018-10-18 22:44:29 -04:00
Minjie Wang	2694b12725	import ffi solution from TVM	2018-09-05 10:51:31 -04:00

9 次代码提交