Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • 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
tensoradapter.h 1.26 KB