• 0
    Refine to_tensor for eager mode and support gpu_pinned (#40535) · 45d1fb8d
    0x45f 提交于
    * Refine to_tensor for eager mode
    
    * support gpu_pinned
    
    * refine code
    
    * support gpu_pinned copy_to
    
    * fix layer.__setattr__
    
    * support to_tensor for gpu_pinned
    
    * fix unit test
    
    * refine gpu_pinned
    
    * restore the original code
    
    * add is_gup_pinned() and refine eager.Tensor._copy_to()
    45d1fb8d
ProcessGroupNCCL.cc 22.3 KB