"scripts/__init__.py" did not exist on "c7ba6ba2678ca7e4e58320da8209be8883a56322"
  • 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
torch.cpp 886 Bytes