• Xin Yao's avatar
    [Feature] Make TensorAdapter Stream Aware (#4472) · 2b766740
    Xin Yao authored
    * Allocate tensors in DGL's current stream
    
    * make tensoradaptor stream-aware
    
    * replace TAemtpy with cpu allocator
    
    * fix typo
    
    * try fix cpu allocation
    
    * clean header
    
    * redirect AllocDataSpace as well
    
    * resolve comments
    2b766740
ndarray.cc 18.1 KB