Fix for failing precision tests

[ankitgola005]

Fix for failing precision tests

Fix precision tests.

• Was this discussed/agreed via a Github issue? (no need for typos and docs improvements)
• Did you read the contributor guideline, 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]

Manual test results

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