• nv-dlasalle's avatar
    [Dataloading] Add class for copying tensors to/from the GPU on a non-default stream (#2284) · f673fc25
    nv-dlasalle authored
    * Add async transferer class
    
    * Add async ndarray copy interface
    
    * Add python bindings
    
    * Fix comment
    
    * Add python class
    
    * Fix linting issues
    
    * Add python unit test
    
    * Update python interface
    
    * move async_transferer to cuda only directory
    
    * Fix linting issue
    
    * Move out of contrib
    
    * Add doc strings
    
    * Move test compute from backend
    
    * Update comment
    
    * Fix test naming
    
    * Fix argument usage
    
    * Wrap/unwrap backend parameters
    
    * Move to dataloading
    
    * Move to 'dataloading'
    
    * Make GPU/CPU compatible
    
    * Fix unit tests
    
    * Add docs
    
    * Use only backend interface for datamovement in unit test
    f673fc25
async_transferer.cc 3 KB