文件历史

作者	SHA1	备注	提交日期
Chang Liu	62b5f50a5c	[Feature] Import PyTorch's `pin_memory()` method for DGL graph structure (#5366 )	2023-04-10 19:21:32 -07:00
Xin Yao	26612d0cb0	[Fix] Fix record_stream for PyTorch 1.14 (#4975 )	2022-12-02 13:31:47 +08:00
Hongzhi (Steve), Chen	bcd3768426	[Misc] Replace /! with /. (#4823 ) replace * blabla * balbla * blabla Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>	2022-11-07 08:16:41 +08:00
Hongzhi (Steve), Chen	619d735df5	[Misc] Replace \xxx with @xxx in structured comment. (#4822 ) * param * brief * note * return * tparam * brief2 * file * return2 * return * blabla * all Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>	2022-11-07 00:22:45 +08:00
Hongzhi (Steve), Chen	33a2d9e1b6	[Misc] clang-format auto fix. (#4812 ) * [Misc] clang-format auto fix. * manual Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>	2022-11-04 14:24:20 +08:00
Xin Yao	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
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	2b76674002	[Feature] Make TensorAdapter Stream Aware (#4472 ) * Allocate tensors in DGL's current stream * make tensoradaptor stream-aware * replace TAemtpy with cpu allocator * fix typo * try fix cpu allocation * clean header * redirect AllocDataSpace as well * resolve comments	2022-08-31 20:09:40 +08:00
Xin Yao	9ee7ced5e3	[Performance] Redirect `AllocWorkspace` to PyTorch's allocator if available (#4199 )	2022-07-07 13:35:44 +08:00
Quan (Andy) Gan	520cef88df	[Feature] Compatibility to DLPack 0.6 in tensoradapter (#3803 ) * compatibility to DLPack 0.6 in tensoradapter * fix * oops Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>	2022-03-10 10:32:52 +08:00
Quan (Andy) Gan	9a7235faf2	[Performance] Use allocator from PyTorch if possible (#2328 ) * 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	2020-12-25 13:57:51 +08:00

11 次代码提交