1. 03 Feb, 2022 1 commit
  2. 28 Jan, 2022 2 commits
  3. 25 Jan, 2022 1 commit
    • Jiarui Fang's avatar
      add pytorch hooks (#179) · 569357fe
      Jiarui Fang authored
      * add pytorch hooks
      fix #175
      
      * remove licenses in src code
      
      * add gpu memory tracer
      
      * replacing print with logger in ophooks.
      569357fe
  4. 21 Jan, 2022 3 commits
  5. 20 Jan, 2022 2 commits
  6. 19 Jan, 2022 6 commits
  7. 18 Jan, 2022 6 commits
  8. 17 Jan, 2022 2 commits
  9. 13 Jan, 2022 1 commit
  10. 10 Jan, 2022 2 commits
  11. 07 Jan, 2022 5 commits
  12. 06 Jan, 2022 2 commits
  13. 05 Jan, 2022 1 commit
  14. 04 Jan, 2022 4 commits
  15. 30 Dec, 2021 2 commits
    • ver217's avatar
      Optimize pipeline schedule (#94) · 96780e6e
      ver217 authored
      
      
      * add pipeline shared module wrapper and update load batch
      
      * added model parallel process group for amp and clip grad (#86)
      
      * added model parallel process group for amp and clip grad
      
      * update amp and clip with model parallel process group
      
      * remove pipeline_prev/next group (#88)
      
      * micro batch offload
      
      * optimize pipeline gpu memory usage
      
      * pipeline can receive tensor shape (#93)
      
      * optimize pipeline gpu memory usage
      
      * fix grad accumulation step counter
      
      * rename classes and functions
      Co-authored-by: default avatarFrank Lee <somerlee.9@gmail.com>
      96780e6e
    • アマデウス's avatar
      added gpt model & benchmark (#95) · e5b9f9a0
      アマデウス authored
      e5b9f9a0