mirrors / dmlc--dgl

文件历史

4,417 提交 34 分支 42 Git 标签
master
提交图

4 次代码提交

作者 SHA1 备注 提交日期
Hongzhi (Steve), Chen a208e8868b [Misc] Black auto fix. (#4680) 
* [Misc] Black auto fix.

* fix pylint disable

Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2022-10-08 14:15:56 +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
Xin Yao 5bef48dfda [Performance] Optimize the use of alternative streams in dataloader (#4177) 
* fix using alternative streams

* use a alternative stream for subgraph transferring

* fix StreamContext when stream is None
 2022-06-29 16:02:25 +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