[XPU] [Test] Make tests in test_float8_tensor.py device agnostic

[Stonepia]

This PR makes test_float8_tensor.py device agnostic and supports XPU.

Dependencies

This PR dependends on following PR.

• PyTorch: [XPU] [Feature] [3/3] Register the scaled_mm and scaled_mm_v2 for xpu pytorch#166056 : 166056 registered torch._scaled_mm for XPU device. Without it, the scaled_mm will fallback to CPU device, and cause the RowWise tests failed (CPU only supports tensorwise scaling for now).

If 166056 not merged, 57 tests will fail with the error message:

Now _scaled_mm only supports per-tensor scaling for CPU backend.

• torchao: [Intel XPU] Enable test/quantization UTs on XPU  #3249 : 3249 adds the util function for auto_detect_device. We will need a rebase after 3249 is merged.

Test Status

After the dependencies in the above are merged, the tests will be 428 passed and 28 skipped. All the skipped tests are due to the fbgemm check _is_fbgemm_gpu_genai_available().

# Skipped Lists
test_bmm: 2 tests
test_expected_gpu_kernel_fbgemm: 1 test.
test_fp8_conv_variants : 16 tests.
test_fp8_conv_skip_quant: 8 tests.
test_moe_weight_reshape_ops: 1 test.

[pytorch-bot]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR

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

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

[Stonepia]

@pytorchbot label "xpu"