1. 01 4月, 2021 2 次提交
    • S
      Support control flow in DataParallel (#31625) · 8460698b
      ShenLiang 提交于
      * support control flow
      
      * supoort sync_parameters_buffers
      
      * fix the bug of sparse embedding
      8460698b
    • C
      Refactor and simplify hook design & add Tensor.register_hook API (#31775) · dbeb3ea4
      Chen Weihang 提交于
      * refactor and simplify hook design
      
      * fix reducer add hook error
      
      * add Tensor.register_hook basic impl
      
      * refine prepare data impl
      
      * revert prepare data change
      
      * support register_hook for Tensor
      
      * add hook test in model
      
      * polish tests and doc example
      
      * fix double grad test failed
      
      * remove reduce hook func
      
      * fix set empty error
      
      * polish code by comments
      
      * change reduce_hook to mutable_hook
      
      * remove useless tmp_ins
      
      * fix shape code format error
      
      * fix shape code format error
      dbeb3ea4
  2. 09 3月, 2021 1 次提交
  3. 05 3月, 2021 1 次提交
  4. 22 2月, 2021 1 次提交
  5. 19 2月, 2021 1 次提交
  6. 09 2月, 2021 1 次提交
  7. 08 2月, 2021 1 次提交
  8. 04 2月, 2021 1 次提交
  9. 03 2月, 2021 1 次提交
  10. 29 1月, 2021 1 次提交
  11. 19 1月, 2021 1 次提交
  12. 13 1月, 2021 1 次提交
  13. 22 12月, 2020 1 次提交
  14. 09 12月, 2020 1 次提交
  15. 03 12月, 2020 1 次提交
  16. 27 11月, 2020 1 次提交
    • S
      Support dynamic graph distributed (#28997) · e2d01eb6
      ShenLiang 提交于
      * add reducer
      
      * refine envent for memorycopy
      
      * add concat&split for allreduce
      
      * apply concat & split for fuse tensor
      
      * fix nccl dep
      
      * fix the untest, compile problem and ddp initialize problem
      
      * fix untest for mac & add some comments & solve the repeated param in sublayers
      
      * fix untest for windows & fix document
      e2d01eb6