• Rhett Ying's avatar
    [Feature] enable async transfer in NodeDataLoader for homograph (#3407) · e053df79
    Rhett Ying authored
    * [Feature] enable async transfer in NodeDataLoader for homograph
    
    * fix lint issues
    
    * fix device choose when creating stream
    
    * fix test on cpu only machine
    
    * fix pin_memory config
    
    * support homo only
    
    * avoid creating stream in each step and sync via event
    
    * fix lint
    
    * enable graph copy on non-default stream
    
    * fix lint
    
    * refine arg description
    
    * fix conflicts
    e053df79
test_dataloader.py 19.1 KB