Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Introduce new precision layout in fabric #16767

Merged
merged 28 commits into from
Feb 17, 2023
Merged

Introduce new precision layout in fabric #16767

merged 28 commits into from
Feb 17, 2023

Conversation

justusschock
Copy link
Member

@justusschock justusschock commented Feb 15, 2023

What does this PR do?

Introduces new precision arguments to fabric.
Also makes sure that plugins only need to support the new version and all conversions happen within the connector.

First part of #9956

Before submitting
  • Was this discussed/agreed via a GitHub issue? (not for typos and docs)
  • Did you read the contributor guideline, Pull Request section?
  • Did you make sure your PR does only one thing, instead of bundling different changes together?
  • Did you make sure to update the documentation with your changes? (if necessary)
  • Did you write any new necessary tests? (not for typos and docs)
  • Did you verify new and existing tests pass locally with your changes?
  • Did you list all the breaking changes introduced by this pull request?
  • Did you update the CHANGELOG? (not for typos, docs, test updates, or minor internal changes/refactors)

PR review

Anyone in the community is welcome to review the PR.
Before you start reviewing, make sure you have read the review guidelines. In short, see the following bullet-list:

Reviewer checklist
  • Is this pull request ready for review? (if not, please submit in draft mode)
  • Check that all items from Before submitting are resolved
  • Make sure the title is self-explanatory and the description concisely explains the PR
  • Add labels and milestones (and optionally projects) to the PR so it can be classified

cc @Borda @carmocca @justusschock @awaelchli

@github-actions github-actions bot added the fabric lightning.fabric.Fabric label Feb 15, 2023
@justusschock justusschock changed the title adapt precision in fabric Introduce new precision layout in fabric Feb 15, 2023
@justusschock justusschock added precision: double Double precision precision: amp Automatic Mixed Precision labels Feb 15, 2023
@justusschock justusschock added this to the 2.0 milestone Feb 15, 2023
@github-actions github-actions bot added the pl Generic label for PyTorch Lightning package label Feb 16, 2023
@justusschock justusschock marked this pull request as ready for review February 16, 2023 09:50
@github-actions
Copy link
Contributor

github-actions bot commented Feb 16, 2023

⚡ Required checks status: All passing 🟢

Groups summary

🟢 pytorch_lightning: Tests workflow
Check ID Status
pl-cpu (macOS-11, lightning, 3.8, 1.11) success
pl-cpu (macOS-11, lightning, 3.9, 1.12) success
pl-cpu (macOS-11, lightning, 3.10, 1.13) success
pl-cpu (macOS-11, lightning, 3.8, 1.11, oldest) success
pl-cpu (ubuntu-20.04, lightning, 3.9, 1.11) success
pl-cpu (ubuntu-20.04, lightning, 3.10, 1.12) success
pl-cpu (ubuntu-20.04, lightning, 3.10, 1.13) success
pl-cpu (ubuntu-20.04, lightning, 3.8, 1.11, oldest) success
pl-cpu (windows-2022, lightning, 3.9, 1.11) success
pl-cpu (windows-2022, lightning, 3.10, 1.12) success
pl-cpu (windows-2022, lightning, 3.10, 1.13) success
pl-cpu (windows-2022, lightning, 3.8, 1.11, oldest) success
pl-cpu (macOS-11, pytorch, 3.8, 1.13) success
pl-cpu (ubuntu-20.04, pytorch, 3.8, 1.13) success
pl-cpu (windows-2022, pytorch, 3.8, 1.13) success

These checks are required after the changes to src/lightning/fabric/cli.py, src/lightning/fabric/connector.py, src/lightning/fabric/fabric.py, src/lightning/fabric/plugins/precision/amp.py, src/lightning/fabric/plugins/precision/deepspeed.py, src/lightning/fabric/plugins/precision/double.py, src/lightning/fabric/plugins/precision/fsdp.py, src/lightning/fabric/plugins/precision/precision.py, src/lightning/fabric/plugins/precision/tpu_bf16.py, src/lightning/fabric/strategies/deepspeed.py, src/lightning/fabric/strategies/fsdp.py, src/lightning/pytorch/plugins/precision/amp.py, src/lightning/pytorch/plugins/precision/deepspeed.py, src/lightning/pytorch/plugins/precision/double.py, src/lightning/pytorch/plugins/precision/hpu.py, src/lightning/pytorch/plugins/precision/ipu.py, src/lightning/pytorch/plugins/precision/tpu_bf16.py, src/lightning/pytorch/trainer/trainer.py.

🟢 pytorch_lightning: Azure GPU
Check ID Status
pytorch-lightning (GPUs) success

These checks are required after the changes to src/lightning/pytorch/plugins/precision/amp.py, src/lightning/pytorch/plugins/precision/deepspeed.py, src/lightning/pytorch/plugins/precision/double.py, src/lightning/pytorch/plugins/precision/hpu.py, src/lightning/pytorch/plugins/precision/ipu.py, src/lightning/pytorch/plugins/precision/tpu_bf16.py, src/lightning/pytorch/trainer/trainer.py, src/lightning/fabric/cli.py, src/lightning/fabric/connector.py, src/lightning/fabric/fabric.py, src/lightning/fabric/plugins/precision/amp.py, src/lightning/fabric/plugins/precision/deepspeed.py, src/lightning/fabric/plugins/precision/double.py, src/lightning/fabric/plugins/precision/fsdp.py, src/lightning/fabric/plugins/precision/precision.py, src/lightning/fabric/plugins/precision/tpu_bf16.py, src/lightning/fabric/strategies/deepspeed.py, src/lightning/fabric/strategies/fsdp.py.

🟢 pytorch_lightning: Azure HPU
Check ID Status
pytorch-lightning (HPUs) success

These checks are required after the changes to src/lightning/fabric/cli.py, src/lightning/fabric/connector.py, src/lightning/fabric/fabric.py, src/lightning/fabric/plugins/precision/amp.py, src/lightning/fabric/plugins/precision/deepspeed.py, src/lightning/fabric/plugins/precision/double.py, src/lightning/fabric/plugins/precision/fsdp.py, src/lightning/fabric/plugins/precision/precision.py, src/lightning/fabric/plugins/precision/tpu_bf16.py, src/lightning/fabric/strategies/deepspeed.py, src/lightning/fabric/strategies/fsdp.py, src/lightning/pytorch/plugins/precision/amp.py, src/lightning/pytorch/plugins/precision/deepspeed.py, src/lightning/pytorch/plugins/precision/double.py, src/lightning/pytorch/plugins/precision/hpu.py, src/lightning/pytorch/plugins/precision/ipu.py, src/lightning/pytorch/plugins/precision/tpu_bf16.py, src/lightning/pytorch/trainer/trainer.py.

🟢 pytorch_lightning: Azure IPU
Check ID Status
pytorch-lightning (IPUs) success

These checks are required after the changes to src/lightning/fabric/cli.py, src/lightning/fabric/connector.py, src/lightning/fabric/fabric.py, src/lightning/fabric/plugins/precision/amp.py, src/lightning/fabric/plugins/precision/deepspeed.py, src/lightning/fabric/plugins/precision/double.py, src/lightning/fabric/plugins/precision/fsdp.py, src/lightning/fabric/plugins/precision/precision.py, src/lightning/fabric/plugins/precision/tpu_bf16.py, src/lightning/fabric/strategies/deepspeed.py, src/lightning/fabric/strategies/fsdp.py, src/lightning/pytorch/plugins/precision/amp.py, src/lightning/pytorch/plugins/precision/deepspeed.py, src/lightning/pytorch/plugins/precision/double.py, src/lightning/pytorch/plugins/precision/hpu.py, src/lightning/pytorch/plugins/precision/ipu.py, src/lightning/pytorch/plugins/precision/tpu_bf16.py, src/lightning/pytorch/trainer/trainer.py.

🟢 pytorch_lightning: Docs
Check ID Status
make-doctest (pytorch) success
make-html (pytorch) success

These checks are required after the changes to src/lightning/pytorch/plugins/precision/amp.py, src/lightning/pytorch/plugins/precision/deepspeed.py, src/lightning/pytorch/plugins/precision/double.py, src/lightning/pytorch/plugins/precision/hpu.py, src/lightning/pytorch/plugins/precision/ipu.py, src/lightning/pytorch/plugins/precision/tpu_bf16.py, src/lightning/pytorch/trainer/trainer.py, docs/source-pytorch/fabric/api/fabric_args.rst, docs/source-pytorch/fabric/fundamentals/launch.rst, docs/source-pytorch/fabric/fundamentals/precision.rst, docs/source-pytorch/fabric/guide/multi_node/cloud.rst.

🟢 lightning_fabric: CPU workflow
Check ID Status
fabric-cpu (macOS-11, lightning, 3.8, 1.11) success
fabric-cpu (macOS-11, lightning, 3.9, 1.12) success
fabric-cpu (macOS-11, lightning, 3.10, 1.13) success
fabric-cpu (macOS-11, lightning, 3.8, 1.11, oldest) success
fabric-cpu (ubuntu-20.04, lightning, 3.9, 1.11) success
fabric-cpu (ubuntu-20.04, lightning, 3.10, 1.12) success
fabric-cpu (ubuntu-20.04, lightning, 3.10, 1.13) success
fabric-cpu (ubuntu-20.04, lightning, 3.8, 1.11, oldest) success
fabric-cpu (windows-2022, lightning, 3.9, 1.11) success
fabric-cpu (windows-2022, lightning, 3.10, 1.12) success
fabric-cpu (windows-2022, lightning, 3.10, 1.13) success
fabric-cpu (windows-2022, lightning, 3.8, 1.11, oldest) success
fabric-cpu (macOS-11, fabric, 3.8, 1.13) success
fabric-cpu (ubuntu-20.04, fabric, 3.8, 1.13) success
fabric-cpu (windows-2022, fabric, 3.8, 1.13) success

These checks are required after the changes to src/lightning/fabric/cli.py, src/lightning/fabric/connector.py, src/lightning/fabric/fabric.py, src/lightning/fabric/plugins/precision/amp.py, src/lightning/fabric/plugins/precision/deepspeed.py, src/lightning/fabric/plugins/precision/double.py, src/lightning/fabric/plugins/precision/fsdp.py, src/lightning/fabric/plugins/precision/precision.py, src/lightning/fabric/plugins/precision/tpu_bf16.py, src/lightning/fabric/strategies/deepspeed.py, src/lightning/fabric/strategies/fsdp.py, tests/tests_fabric/plugins/precision/test_amp.py, tests/tests_fabric/plugins/precision/test_amp_integration.py, tests/tests_fabric/plugins/precision/test_deepspeed.py, tests/tests_fabric/plugins/precision/test_deepspeed_integration.py, tests/tests_fabric/plugins/precision/test_double_integration.py, tests/tests_fabric/plugins/precision/test_fsdp.py, tests/tests_fabric/strategies/test_deepspeed_integration.py, tests/tests_fabric/strategies/test_fsdp_integration.py, tests/tests_fabric/test_cli.py, tests/tests_fabric/test_connector.py.

🟢 lightning_fabric: Azure GPU
Check ID Status
lightning-fabric (GPUs) success

These checks are required after the changes to src/lightning/fabric/cli.py, src/lightning/fabric/connector.py, src/lightning/fabric/fabric.py, src/lightning/fabric/plugins/precision/amp.py, src/lightning/fabric/plugins/precision/deepspeed.py, src/lightning/fabric/plugins/precision/double.py, src/lightning/fabric/plugins/precision/fsdp.py, src/lightning/fabric/plugins/precision/precision.py, src/lightning/fabric/plugins/precision/tpu_bf16.py, src/lightning/fabric/strategies/deepspeed.py, src/lightning/fabric/strategies/fsdp.py, tests/tests_fabric/plugins/precision/test_amp.py, tests/tests_fabric/plugins/precision/test_amp_integration.py, tests/tests_fabric/plugins/precision/test_deepspeed.py, tests/tests_fabric/plugins/precision/test_deepspeed_integration.py, tests/tests_fabric/plugins/precision/test_double_integration.py, tests/tests_fabric/plugins/precision/test_fsdp.py, tests/tests_fabric/strategies/test_deepspeed_integration.py, tests/tests_fabric/strategies/test_fsdp_integration.py, tests/tests_fabric/test_cli.py, tests/tests_fabric/test_connector.py.

🟢 mypy
Check ID Status
mypy success

These checks are required after the changes to src/lightning/fabric/cli.py, src/lightning/fabric/connector.py, src/lightning/fabric/fabric.py, src/lightning/fabric/plugins/precision/amp.py, src/lightning/fabric/plugins/precision/deepspeed.py, src/lightning/fabric/plugins/precision/double.py, src/lightning/fabric/plugins/precision/fsdp.py, src/lightning/fabric/plugins/precision/precision.py, src/lightning/fabric/plugins/precision/tpu_bf16.py, src/lightning/fabric/strategies/deepspeed.py, src/lightning/fabric/strategies/fsdp.py, src/lightning/pytorch/plugins/precision/amp.py, src/lightning/pytorch/plugins/precision/deepspeed.py, src/lightning/pytorch/plugins/precision/double.py, src/lightning/pytorch/plugins/precision/hpu.py, src/lightning/pytorch/plugins/precision/ipu.py, src/lightning/pytorch/plugins/precision/tpu_bf16.py, src/lightning/pytorch/trainer/trainer.py.

🟢 install
Check ID Status
install-pkg (ubuntu-22.04, app, 3.8) success
install-pkg (ubuntu-22.04, app, 3.10) success
install-pkg (ubuntu-22.04, fabric, 3.8) success
install-pkg (ubuntu-22.04, fabric, 3.10) success
install-pkg (ubuntu-22.04, pytorch, 3.8) success
install-pkg (ubuntu-22.04, pytorch, 3.10) success
install-pkg (ubuntu-22.04, lightning, 3.8) success
install-pkg (ubuntu-22.04, lightning, 3.10) success
install-pkg (ubuntu-22.04, notset, 3.8) success
install-pkg (ubuntu-22.04, notset, 3.10) success
install-pkg (macOS-12, app, 3.8) success
install-pkg (macOS-12, app, 3.10) success
install-pkg (macOS-12, fabric, 3.8) success
install-pkg (macOS-12, fabric, 3.10) success
install-pkg (macOS-12, pytorch, 3.8) success
install-pkg (macOS-12, pytorch, 3.10) success
install-pkg (macOS-12, lightning, 3.8) success
install-pkg (macOS-12, lightning, 3.10) success
install-pkg (macOS-12, notset, 3.8) success
install-pkg (macOS-12, notset, 3.10) success
install-pkg (windows-2022, app, 3.8) success
install-pkg (windows-2022, app, 3.10) success
install-pkg (windows-2022, fabric, 3.8) success
install-pkg (windows-2022, fabric, 3.10) success
install-pkg (windows-2022, pytorch, 3.8) success
install-pkg (windows-2022, pytorch, 3.10) success
install-pkg (windows-2022, lightning, 3.8) success
install-pkg (windows-2022, lightning, 3.10) success
install-pkg (windows-2022, notset, 3.8) success
install-pkg (windows-2022, notset, 3.10) success

These checks are required after the changes to src/lightning/fabric/cli.py, src/lightning/fabric/connector.py, src/lightning/fabric/fabric.py, src/lightning/fabric/plugins/precision/amp.py, src/lightning/fabric/plugins/precision/deepspeed.py, src/lightning/fabric/plugins/precision/double.py, src/lightning/fabric/plugins/precision/fsdp.py, src/lightning/fabric/plugins/precision/precision.py, src/lightning/fabric/plugins/precision/tpu_bf16.py, src/lightning/fabric/strategies/deepspeed.py, src/lightning/fabric/strategies/fsdp.py, src/lightning/pytorch/plugins/precision/amp.py, src/lightning/pytorch/plugins/precision/deepspeed.py, src/lightning/pytorch/plugins/precision/double.py, src/lightning/pytorch/plugins/precision/hpu.py, src/lightning/pytorch/plugins/precision/ipu.py, src/lightning/pytorch/plugins/precision/tpu_bf16.py, src/lightning/pytorch/trainer/trainer.py.

🟢 link-check
Check ID Status
markdown-link-check success

These checks are required after the changes to src/lightning/fabric/CHANGELOG.md.


Thank you for your contribution! 💜

Note
This comment is automatically generated and updates for 60 minutes every 180 seconds. If you have any other questions, contact carmocca for help.

@justusschock justusschock marked this pull request as draft February 16, 2023 10:57
@justusschock justusschock marked this pull request as ready for review February 16, 2023 12:03
@codecov
Copy link

codecov bot commented Feb 16, 2023

Codecov Report

Merging #16767 (3bdd879) into master (1a6331f) will decrease coverage by 22%.
The diff coverage is 86%.

Additional details and impacted files
@@            Coverage Diff            @@
##           master   #16767     +/-   ##
=========================================
- Coverage      82%      59%    -22%     
=========================================
  Files         441      416     -25     
  Lines       31654    31371    -283     
=========================================
- Hits        25816    18595   -7221     
- Misses       5838    12776   +6938     

Copy link
Contributor

@awaelchli awaelchli left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Tried hard to find a typo but couldn't!

src/lightning/fabric/plugins/precision/precision.py Outdated Show resolved Hide resolved
src/lightning/fabric/connector.py Outdated Show resolved Hide resolved
tests/tests_fabric/test_connector.py Show resolved Hide resolved
tests/tests_fabric/test_connector.py Outdated Show resolved Hide resolved
src/lightning/fabric/connector.py Outdated Show resolved Hide resolved
src/lightning/fabric/connector.py Outdated Show resolved Hide resolved
src/lightning/fabric/plugins/precision/amp.py Show resolved Hide resolved
@mergify mergify bot added the ready PRs ready to be merged label Feb 16, 2023
@justusschock justusschock enabled auto-merge (squash) February 17, 2023 10:01
@justusschock justusschock merged commit ac5fa03 into master Feb 17, 2023
@justusschock justusschock deleted the 2.0/precision branch February 17, 2023 10:41
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
fabric lightning.fabric.Fabric pl Generic label for PyTorch Lightning package precision: amp Automatic Mixed Precision precision: double Double precision ready PRs ready to be merged
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants