1. 09 6月, 2022 1 次提交
  2. 07 6月, 2022 1 次提交
  3. 05 6月, 2022 1 次提交
  4. 28 4月, 2022 1 次提交
  5. 07 4月, 2022 1 次提交
  6. 04 3月, 2022 1 次提交
  7. 02 3月, 2022 1 次提交
  8. 01 3月, 2022 1 次提交
  9. 25 2月, 2022 1 次提交
  10. 20 2月, 2022 1 次提交
  11. 19 2月, 2022 1 次提交
    • S
      Add the DistributedFusedLamb optimizer (#39148) · 5df3cd61
      sneaxiy 提交于
      * add DistributedFusedLamb op
      
      * polish code
      
      * fix compile error
      
      * compatible with pten changement
      
      * fix rocm compile error
      
      * improve converage
      
      * update upstream/develop
      
      * fix cast_with_ptr.h
      
      * add FLAGS_distributed_lamb_divide_nranks_when_allreduce=1
      
      * fix clip before allreduce
      
      * add use_master_param_norm
      
      * code polish
      
      * fix bug
      
      * fix ROCM ci
      5df3cd61