1. 17 10月, 2022 1 次提交
    • W
      [Cherry-pick] Collective communication APIs (#46922) · 5fba2a98
      Wen Sun 提交于
      * Support both use_calc_stream and sync_op in send recv APIs (#46023)
      
      * Support both use_calc_stream and sync_op in allgather API (#46295)
      
      * Support both use_calc_stream and sync_op in collective communication API (#46761)
      
      * Move group and all reduce from collective to communication (#45848)
      
      * Completes bfloat16 dtype for collective api in eager mode (#45844)
      
      * Fix collective APIs cannot be recognized when building docs (#46962)
      Co-authored-by: NLiYuRio <63526175+LiYuRio@users.noreply.github.com>
      5fba2a98
  2. 07 9月, 2022 1 次提交
  3. 06 9月, 2022 1 次提交
  4. 31 8月, 2022 1 次提交
  5. 12 8月, 2022 1 次提交
  6. 08 8月, 2022 1 次提交
  7. 01 8月, 2022 1 次提交
    • L
      unify gpu context (#44740) · 86763023
      Leo Chen 提交于
      * remove cudaDeviceContext
      
      * remove more template
      
      * fix rocm compile
      
      * remove alias name CUDADeviceContext
      
      * fix compile
      
      * fix tests
      
      * revert changes
      86763023
  8. 22 7月, 2022 1 次提交
  9. 11 7月, 2022 1 次提交
  10. 26 6月, 2022 1 次提交
  11. 13 6月, 2022 1 次提交
  12. 05 6月, 2022 1 次提交
  13. 28 5月, 2022 1 次提交
  14. 19 4月, 2022 1 次提交
  15. 18 4月, 2022 1 次提交
  16. 13 4月, 2022 1 次提交
  17. 07 4月, 2022 2 次提交
  18. 31 3月, 2022 1 次提交
  19. 22 3月, 2022 1 次提交
  20. 21 3月, 2022 1 次提交
    • 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
  21. 15 3月, 2022 1 次提交
  22. 07 3月, 2022 1 次提交
  23. 03 3月, 2022 1 次提交
  24. 02 3月, 2022 1 次提交
  25. 23 2月, 2022 1 次提交