项目文件夹

文件

T

文件历史

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

_ctypes

[Bugfix] Bug fixes in new dataloader (#3727 )

2022-02-22 01:30:31 +08:00

_cy2

import ffi solution from TVM

2018-09-05 10:51:31 -04:00

_cy3

import ffi solution from TVM

2018-09-05 10:51:31 -04:00

_cython

[Feature] Import PyTorch's CUDA stream management (#4503 )