[distributed] error message does not match

[daisyden]

🐛 Describe the bug

Unit test test.distributed._composable.fsdp.test_fully_shard_state_dict.TestFullyShardStateDictMultiProcess | test_dp_state_dict_cpu_offload got assertion:

AssertionError: "Found following parameters on non-CPU device: [('0.weight', device(type={device_type}" does not match "FSDP parameters should be materialized on CPU when enabling CPU offloading. For example, load a CPU state dict or call module.to_empty(device="cpu"). Found following parameters on non-CPU device: [('0.weight', device(type='xpu', index=7))]

Versions

2025.1 oneapi
2025.15 oneCCL
4 XELINK 1110

[zhangxiaoli73]

@daisyden It seems you have not enabled this test correctly. Please check with developer branch and fix.

[daisyden]

Fixed and checked into distributed_2.8.