https://github.com/dmlc/dgl
Revision 960092be02bfae2a32a300081f928d60f23e7246 authored by Xin Yao on 15 September 2022, 08:54:52 UTC, committed by Quan Gan on 16 September 2022, 02:50:27 UTC
* 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
1 parent 75d23de
Tip revision: 960092be02bfae2a32a300081f928d60f23e7246 authored by Xin Yao on 15 September 2022, 08:54:52 UTC
[Feature] Import PyTorch's CUDA stream management (#4503)
[Feature] Import PyTorch's CUDA stream management (#4503)
Tip revision: 960092b
| File | Mode | Size |
|---|---|---|
| .github | ||
| apps | ||
| benchmarks | ||
| cmake | ||
| conda | ||
| dglgo | ||
| docker | ||
| docs | ||
| examples | ||
| featgraph | ||
| include | ||
| python | ||
| src | ||
| tensoradapter | ||
| tests | ||
| third_party | ||
| tools | ||
| tutorials | ||
| .gitignore | -rw-r--r-- | 2.4 KB |
| .gitmodules | -rw-r--r-- | 1.2 KB |
| CMakeLists.txt | -rw-r--r-- | 12.8 KB |
| CONTRIBUTORS.md | -rw-r--r-- | 3.5 KB |
| Jenkinsfile | -rw-r--r-- | 19.9 KB |
| LICENSE | -rw-r--r-- | 11.1 KB |
| NEWS.md | -rw-r--r-- | 2.1 KB |
| README.md | -rw-r--r-- | 30.9 KB |
| readthedocs.yml | -rw-r--r-- | 119 bytes |
Computing file changes ...
