1. 12 1月, 2023 1 次提交
    • J
      Fix reduce func bug in process_group_bkcl (#49749) · 8e291bf7
      jameszhang 提交于
      * Fix reduce func bug in process_group_bkcl
      
      Also catch up with a recent process_group PR that failed to add XPU branch.
      Note that reduce is still accomplished by allreduce for xpu. Fix this should
      xccl lib be updated.
      
      * fix compile issue for non-XPU
      8e291bf7
  2. 09 1月, 2023 1 次提交
  3. 06 1月, 2023 1 次提交
  4. 05 1月, 2023 1 次提交
  5. 19 12月, 2022 1 次提交
  6. 17 12月, 2022 1 次提交
  7. 16 12月, 2022 1 次提交
  8. 15 12月, 2022 1 次提交
  9. 14 12月, 2022 1 次提交
  10. 12 12月, 2022 1 次提交
  11. 05 12月, 2022 1 次提交
  12. 03 12月, 2022 1 次提交
  13. 24 11月, 2022 1 次提交
  14. 23 11月, 2022 1 次提交
  15. 21 11月, 2022 4 次提交
  16. 19 11月, 2022 1 次提交
  17. 18 11月, 2022 3 次提交
    • W
    • J
      correct sync behavior for XPU distributed training (#47882) · aafa9820
      james 提交于
      * correct sync behavior for XPU distributed training
      
      XPU support event mechanism similar to cuda event, so it is advisable to
      use an event to sync compute/comm streams for performance. However this
      mechanism is never fully tested, and inconsistent loss/ending_epochs are
      reported. Therefore, this PR replaces event sync with stream waiting as
      a temporary solution.
      
      * remove compile warning
      aafa9820
    • J
      fix device id issue for xpu eager mode (#48076) · 3b18d96b
      james 提交于
      * fix device id issue for xpu eager
      
      xpu device id is not correctly set in eager mode, thus vars are on dev0 unless
      XPUDeviceGurad is called, leading to this error message for all node rank != 0:
      "NotImplementedError: (Unimplemented) Place Place(xpu:0) is not supported."
      
      * fix typo
      
      * fix pybind error
      3b18d96b
  18. 17 11月, 2022 1 次提交
  19. 16 11月, 2022 1 次提交
  20. 14 11月, 2022 3 次提交
  21. 10 11月, 2022 2 次提交
    • J
      XPU multi-card support eager mode (#47445) · 3b91f8f3
      james 提交于
      * XPU support eager mode
      
      * add unittest for XPU eager mode
      
      * minor bugfix
      
      * minor bugfix, test=kunlun
      
      * correct copyright info
      
      * 1. remove unsed vars/funcs
      2. ProcessGroupBKCL inherit from ProcessGroupStream
      
      * bugfix for fp16 in eager mode multi-card, test=kunlun
      
      * rebase & fix a few issues
      
      * use new processgroup interface, test=kunlun
      
      * fix compile issue, test=kunlun
      3b91f8f3
    • W
      Refactor collective communication P2P C++ API (#47801) · d926c270
      Wen Sun 提交于
      * refactor: send, recv, send_partial, recv_partial
      
      * refactor: rm useless const ref
      d926c270
  22. 09 11月, 2022 1 次提交
  23. 08 11月, 2022 1 次提交
  24. 07 11月, 2022 1 次提交
  25. 04 11月, 2022 2 次提交
  26. 01 11月, 2022 1 次提交
  27. 31 10月, 2022 1 次提交
  28. 28 10月, 2022 2 次提交
  29. 17 10月, 2022 1 次提交
    • G
      Support BF16 training for sharding (#46846) · 0b39b244
      Ghost Screaming 提交于
      * Fix bug of reduce_sum op. When input.numel() > INT32_MAX, its result
      is wrong.
      
      * support pure bfloat16
      
      * support bf16 linear
      
      * update PR to pass CI
      
      * tiny fix where_grad_kernel.cu
      
      * Support bfloat16 type for reducer and sharding.
      
      * Fix some bug.
      
      * Polish code.
      
      * Polise code.
      
      * Add bfloat16 datatype in fill_grad kernels.
      Co-authored-by: Nsneaxiy <sneaxiy@126.com>
      0b39b244
  30. 11 10月, 2022 1 次提交