search

Lightning-AI / lightning-Habana

Lightning support for Intel Habana accelerators.
Apache License 2.0
25 stars 8 forks source link

Add checkpointing tests #252

Open ankitgola005 opened 3 weeks ago

ankitgola005 commented 3 weeks ago
Before submitting - [ ] Was this discussed/agreed via a Github issue? (no need for typos and docs improvements) - [ ] Did you read the [contributor guideline](https://github.com/Lightning-AI/pytorch-lightning/blob/main/.github/CONTRIBUTING.md), Pull Request section? - [ ] Did you make sure to update the docs? - [ ] Did you write any new necessary tests?

What does this PR do?

Fixes # (issue).

PR review

Anyone in the community is free to review the PR once the tests have passed. If we didn't discuss your PR in GitHub issues there's a high chance it will not be merged.

Did you have fun?

Make sure you had fun coding 🙃

ankitgola005 commented 2 weeks ago

Manual testing results. Note: To be rebased over #251.

Category pytorch-lightning \ pypi   pytorch-lightning \ source   lightning \ pypi   lightning \ source  
  G1 G2 G1 G2 G1 G2 G1 G2
HPU General tests P P P P P P P P
HPU torch compile tests P P P P P P P P
HPU FSDP tests P P P P P P P P
HPU Deepspeed tests P P P P P P P P
HPU precision test P P P P P P P P
Standalone-only single card tests P P P P P P P P
Multi card(2) HPU test P P P P P P P P
FSDP PT Multi card(2) HPU test P P P P P P P P
FSDP Fabric Multi card(2) HPU test P P P P P P P P
Testing HPU examples P P P P P P P P