DeepSpeed
9caa74e5 - Autotuning (#1554)

Commit

3 years ago

Autotuning (#1554) * [squash] Staging autotuning v4 Co-authored-by: Cheng Li <pistasable@gmail.com> Co-authored-by: Minjia Zhang <minjiaz@microsoft.com> Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com> Co-authored-by: Jeff Rasley <jerasley@microsoft.com> * add new extra, guard xgboost, cleanup dead files (#268) * Fix autotuning docs (#1553) * fix docs * rewording the goal * fix typos * fix typos (#1556) * fix typos * fix format * fix bug (#1557) * fix bug Co-authored-by: Jeff Rasley <jerasley@microsoft.com> Co-authored-by: Minjia Zhang <minjiaz@microsoft.com> Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>

References

#1554 - Autotuning

Author

cli99

Parents

b7cc7c8e

Files46

.github/workflows
- formatting.yml
- main.yml
MANIFEST.in
deepspeed
- autotuning
  - .gitignore
  - README.md
  - __init__.py
  - autotuner.py
  - config.py
  - config_templates
    - template_zero0.json
    - template_zero1.json
    - template_zero2.json
    - template_zero3.json
  - constants.py
  - scheduler.py
  - tuner
    - README.md
    - __init__.py
    - base_tuner.py
    - cost_model.py
    - index_based_tuner.py
    - model_based_tuner.py
    - utils.py
  - utils.py
- launcher
  - runner.py
- runtime
  - config.py
  - config_utils.py
  - engine.py
  - utils.py
  - zero
    - offload_config.py
    - offload_constants.py
    - partition_parameters.py
    - stage2.py
- utils
  - logging.py
  - timer.py
docs
- _config.yml
- _data
  - navigation.yml
- _pages
  - config-json.md
  - features.md
- _tutorials
  - autotuning.md
- code-docs/source
  - autotuning.rst
  - index.rst
requirements
- requirements-autotuning-ml.txt
- requirements-autotuning.txt
- requirements.txt
setup.py
tests/unit
- test_autotuning.py
- test_zero.py

DeepSpeed 9caa74e5 - Autotuning (#1554)

DeepSpeed
9caa74e5 - Autotuning (#1554)