1. 05 Jun, 2023 1 commit
    • Hongxin Liu's avatar
      [bf16] add bf16 support (#3882) · ae02d4e4
      Hongxin Liu authored
      * [bf16] add bf16 support for fused adam (#3844)
      
      * [bf16] fused adam kernel support bf16
      
      * [test] update fused adam kernel test
      
      * [test] update fused adam test
      
      * [bf16] cpu adam and hybrid adam optimizers support bf16 (#3860)
      
      * [bf16] implement mixed precision mixin and add bf16 support for low level zero (#3869)
      
      * [bf16] add mixed precision mixin
      
      * [bf16] low level zero optim support bf16
      
      * [text] update low level zero test
      
      * [text] fix low level zero grad acc test
      
      * [bf16] add bf16 support for gemini (#3872)
      
      * [bf16] gemini support bf16
      
      * [test] update gemini bf16 test
      
      * [doc] update gemini docstring
      
      * [bf16] add bf16 support for plugins (#3877)
      
      * [bf16] add bf16 support for legacy zero (#3879)
      
      * [zero] init context support bf16
      
      * [zero] legacy zero support bf16
      
      * [test] add zero bf16 test
      
      * [doc] add bf16 related docstring for legacy zero
      ae02d4e4
  2. 06 Apr, 2023 1 commit
  3. 04 Apr, 2023 1 commit
    • ver217's avatar
      [zero] reorganize zero/gemini folder structure (#3424) · 26b7aac0
      ver217 authored
      * [zero] refactor low-level zero folder structure
      
      * [zero] fix legacy zero import path
      
      * [zero] fix legacy zero import path
      
      * [zero] remove useless import
      
      * [zero] refactor gemini folder structure
      
      * [zero] refactor gemini folder structure
      
      * [zero] refactor legacy zero import path
      
      * [zero] refactor gemini folder structure
      
      * [zero] refactor gemini folder structure
      
      * [zero] refactor gemini folder structure
      
      * [zero] refactor legacy zero import path
      
      * [zero] fix test import path
      
      * [zero] fix test
      
      * [zero] fix circular import
      
      * [zero] update import
      26b7aac0
  4. 06 Jan, 2023 1 commit
  5. 04 Jan, 2023 1 commit
  6. 23 Dec, 2022 4 commits
  7. 12 Dec, 2022 1 commit
  8. 17 Nov, 2022 1 commit
  9. 05 Aug, 2022 1 commit
  10. 26 Jul, 2022 1 commit
    • ver217's avatar
      [nvme] CPUAdam and HybridAdam support NVMe offload (#1360) · c415240d
      ver217 authored
      * impl nvme optimizer
      
      * update cpu adam
      
      * add unit test
      
      * update hybrid adam
      
      * update docstr
      
      * add TODOs
      
      * update CI
      
      * fix CI
      
      * fix CI
      
      * fix CI path
      
      * fix CI path
      
      * fix CI path
      
      * fix install tensornvme
      
      * fix CI
      
      * fix CI path
      
      * fix CI env variables
      
      * test CI
      
      * test CI
      
      * fix CI
      
      * fix nvme optim __del__
      
      * fix adam __del__
      
      * fix nvme optim
      
      * fix CI env variables
      
      * fix nvme optim import
      
      * test CI
      
      * test CI
      
      * fix CI
      c415240d
  11. 25 Mar, 2022 2 commits
  12. 16 Mar, 2022 1 commit
  13. 11 Mar, 2022 1 commit