1. 30 Mar, 2022 1 commit
  2. 08 Mar, 2022 1 commit
  3. 13 Jan, 2022 1 commit
    • Anupam Bhatnagar's avatar
      [feature] [experimental] Layerwise Gradient Scaler (#879) · 52d066a2
      Anupam Bhatnagar authored
      * [skip ci] first commit
      
      * [skip ci] gradient scaler example
      
      * [skip ci] adding feed forward toy example
      
      * [skip ci] adding types
      
      * [skip ci] adding backward hook
      
      * [skip ci] update
      
      * [skip ci] working feed forward example
      
      * [skip ci] working feed forward example
      
      * [skip ci] use named_modules instead of named_children
      
      * [skip ci] adding new file
      
      * [skip ci] clean up
      
      * [skip ci] implement unscale function
      
      * [skip ci] implement unscale function
      
      * [skip ci] removing old file
      
      * [skip ci] removing some more old files
      
      * [skip ci] making unscale function generic
      
      * [skip ci] adding test for vision model
      
      * [skip ci] adding identity layer
      
      * [skip ci] cleanup files
      
      * [skip ci] refactoring
      
      * [skip ci] more refactoring
      
      * [skip ci] added functionality to update scale
      
      * [skip ci] data loader clean up
      
      * [skip ci] implemented inf checks and update scale functions
      
      * [skip ci]code clean up. added test with autocast. does not work atm
      
      * adding documentation
      
      * adding dependency in requirements-dev.txt
      
      * updating pytorch nightly version
      
      * updating changelog
      
      * adding is_cuda_available to test_vision_model
      
      * set same timeout on cpu and gpu
      
      * reverting cpu timeout, skip vision test on cpu
      
      * addressing comments, fixing vision test
      
      * unscale uses in-place matmul
      
      * some more cleanup
      52d066a2
  4. 24 Nov, 2021 1 commit
    • Ying Zhang's avatar
      [benchmarks]Add an MOE benchmark (#866) · 56254247
      Ying Zhang authored
      * Add MOE to lm benchmarks
      
      * linter
      
      * Fix source / target
      
      * address comments
      
      * address comments
      
      * address comments
      
      * add circleci
      
      * fix circleci
      
      * precommit
      56254247
  5. 18 Nov, 2021 1 commit
  6. 17 Nov, 2021 1 commit
  7. 12 Nov, 2021 1 commit
    • Anupam Bhatnagar's avatar
      Setup pre-commit github action and apply pre-commit to all files (#849) · 7d7edf6d
      Anupam Bhatnagar authored
      * adding pre-commit files
      
      * applying pre-commit to all files
      
      * adding no-strict-optional argument to mypy in circle ci config
      
      * fix typo
      
      * updating python versions
      
      * [skip ci] remove extra args
      
      * adding python 3.9
      
      * [skip ci] set pre-commit version in requirements-dev.txt
      
      * set CACHE_VERSION
      
      * move linters from circleci to github actions
      
      * update python version
      
      * update python version in benchmarks_2
      
      * moving to python 3.9.7
      7d7edf6d
  8. 08 Nov, 2021 1 commit
  9. 10 Aug, 2021 1 commit
    • Rahul Iyer's avatar
      Fix pre-commit hook failures (#756) · 31d600cc
      Rahul Iyer authored
      Pre-commit hook fails when run on all files for three reasons:
      (see trace below)
      
      1. Trailing whitespace on multiple files
      2. mypy fails to load numpy and then subsequently fails to load
      LazyModule from pipe.py
      3. isort sees issues with known_third_party packages
      
      ```
      > pre-commit run --all-files
      
      Trim Trailing Whitespace.................................................Failed
      - hook id: trailing-whitespace
      - exit code: 1
      - files were modified by this hook
      
      Fixing docs/source/conf.py
      Fixing fairscale/experimental/nn/auto_shard.py
      Fixing docs/source/deep_dive/activation_checkpointing.rst
      Fixing docs/source/tutorials/pipe.rst
      Fixing docs/source/installation_instructions.rst
      Fixing docs/source/deep_dive/pipeline_parallelism.rst
      Fixing docs/source/tutorials/activation_checkpointing.rst
      Fixing docs/source/tutorials/offload_model.rst
      Fixing docs/source/deep_dive/oss_sdp_fsdp.rst
      Fixing docs/source/what_is_fairscale.rst
      Fixing CHANGELOG.md
      Fixing fairscale/experimental/nn/offload.py
      Fixing docs/source/index.rst
      Fixing docs/source/deep_dive/adascale.rst
      Fixing README.md
      Fixing docs/source/tutorials/oss.rst
      Fixing docs/source/deep_dive/offload.rst
      
      Check python ast.........................................................Passed
      Check for merge conflicts................................................Passed
      Don't commit to branch...................................................Passed
      Check for added large files..............................................Passed
      Fix End of Files.........................................................Failed
      - hook id: end-of-file-fixer
      - exit code: 1
      - files were modified by this hook
      
      Fixing requirements.txt
      Fixing docs/source/getting_started.rst
      Fixing docs/source/installation_instructions.rst
      Fixing codecov.yml
      Fixing docs/source/deep_dive/adascale.rst
      Fixing docs/source/tutorials/oss.rst
      Fixing docs/source/deep_dive/offload.rst
      
      black....................................................................Passed
      flake8...................................................................Passed
      seed isort known_third_party.............................................Failed
      - hook id: seed-isort-config
      - exit code: 1
      - files were modified by this hook
      isort....................................................................Passed
      mypy.....................................................................Failed
      - hook id: mypy
      - exit code: 2
      
      setup.cfg:45: error: Error importing plugin 'numpy.typing.mypy_plugin': No module named 'numpy'
      Found 1 error in 1 file (checked 197 source files)
      ```
      31d600cc
  10. 28 Apr, 2021 1 commit
  11. 23 Feb, 2021 1 commit
  12. 10 Feb, 2021 1 commit
  13. 01 Feb, 2021 2 commits
  14. 26 Jan, 2021 1 commit
  15. 20 Jan, 2021 1 commit
  16. 11 Jan, 2021 1 commit
  17. 08 Jan, 2021 1 commit
  18. 16 Dec, 2020 1 commit
  19. 01 Dec, 2020 1 commit
  20. 10 Nov, 2020 1 commit
    • Tom Birch's avatar
      Single-process control via PipeRPCWrapper (#156) · 5d4f50fb
      Tom Birch authored
      Adds support for:
      * Reused layers (e.g. for weight sharing)
      * Lazily-constructed layers
      * Single-process control via PipeRPCWrapper
      * PipelineStyle.AsyncScheudle, which lays the foundation for asynchronous pipeline work by introducing an event loop for each rank/worker to process either activations or gradients as they arrive
      
      Also added examples for multi-process and PipeRPCWrapper
      5d4f50fb
  21. 28 Oct, 2020 1 commit
  22. 26 Oct, 2020 1 commit
  23. 09 Sep, 2020 1 commit
  24. 31 Jul, 2020 1 commit
  25. 08 Jul, 2020 1 commit