1. 25 11月, 2022 2 次提交
  2. 21 11月, 2022 1 次提交
  3. 17 11月, 2022 2 次提交
  4. 14 11月, 2022 1 次提交
  5. 10 11月, 2022 1 次提交
  6. 08 11月, 2022 1 次提交
  7. 05 11月, 2022 1 次提交
  8. 04 11月, 2022 1 次提交
  9. 03 11月, 2022 2 次提交
  10. 02 11月, 2022 2 次提交
  11. 01 11月, 2022 1 次提交
  12. 23 10月, 2022 1 次提交
  13. 21 10月, 2022 1 次提交
  14. 20 10月, 2022 3 次提交
  15. 19 10月, 2022 1 次提交
  16. 17 10月, 2022 2 次提交
    • G
      Add enable_partial_send_recv switch in pipeline_configs (#46992) · b9a2f29c
      Ghost Screaming 提交于
      * Fix bug of reduce_sum op. When input.numel() > INT32_MAX, its result
      is wrong.
      
      * Support allow_partial switch, which can be configure in
      pipeline_configs. If sent tensor are not the same from
      different hosts, they shouldn't been sent partially and
      then concated as a whole tensor.
      
      * Change name allow_partial to enable_partial_send_recv.
      
      * Add global variable _enable_partial_send_recv
      b9a2f29c
    • 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
  17. 13 10月, 2022 1 次提交
  18. 12 10月, 2022 2 次提交
  19. 09 10月, 2022 1 次提交
  20. 08 10月, 2022 1 次提交
  21. 28 9月, 2022 2 次提交
  22. 22 9月, 2022 1 次提交
  23. 21 9月, 2022 1 次提交
  24. 20 9月, 2022 2 次提交
  25. 19 9月, 2022 1 次提交
  26. 16 9月, 2022 1 次提交
    • W
      refactor mp. (#45803) · fa97e5ba
      wuhuachaocoding 提交于
      * refactor mp.
      
      * update setup.py.
      
      * update mp_layers.py for compatibility.
      
      * add documents for mp_layers.py
      
      * update init.py
      
      * update collective.py.
      
      * update.
      
      * update mp_ops.py
      
      * update.
      
      * update code style.
      
      * update code style.
      fa97e5ba
  27. 14 9月, 2022 1 次提交
  28. 09 9月, 2022 2 次提交
  29. 07 9月, 2022 1 次提交