• Z
    [Unify Tensors PR #1] Replaced pten::Allocation with... · 42cf2bee
    Zhanlue Yang 提交于
    [Unify Tensors PR #1] Replaced pten::Allocation with shared_ptr<memory::Allocation> for Storage (#38301)
    
    * Added shared_ptr<Allocation> member & corresponding interfaces to Storage
    
    * Removed original pten::Allocation from Storage and adjusted the interfaces accordingly
    
    * Fixed issues with storage offset
    
    * Used place to malloc allocation for TensorStorage
    42cf2bee
allocator.h 1.6 KB