[xpu][float8] Enable float8 fsdp2 tests for Intel XPU

[arlesniak]

This PR enables float8 fsdp2 for Intel XPU tests as part of the #3576.

Details:

• The PR does not add any new test logic.
• It introduces a device parameter to allow running the tests on XPU.
• Source code modifications fix failing tests.

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/ao/4510

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

❌ 1 New Failure, 1 Unrelated Failure

As of commit 3ad5700 with merge base ede5632 ():

NEW FAILURE - The following job has failed:

• PR Label Check / Check PR Labels (gh)
  Process completed with exit code 1.

FLAKY - The following job failed but was likely due to flakiness present on trunk:

• xpu-test / test (gh) (detected as infra flaky with no log or failing log classifier)

This comment was automatically generated by Dr. CI and updates every 15 minutes.

[pytorch-bot]

Workflows were awaiting approval. CI has now been triggered for the ciflow labels on this PR.