• Xin Yao's avatar
    [Feature] Import PyTorch's CUDA stream management (#4503) · 9a00cf19
    Xin Yao authored
    * 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
    9a00cf19
test_unit_graph.cc 14.9 KB