Remove out-dated xpu device check code in get_balanced_memory

[faaany]

What does this PR do?

This PR removes the outdated and unnecessary code of xpu device check in the following code snippet

if (
                    d != "cpu"
                    and (torch.device(d).type == "xpu" or torch.xpu.get_device_properties(d).dev_type == "gpu")
                )

The reasons are

• If torch.xpu.get_device_properties(d) works, torch.device(d).type will always be "xpu"
• If torch.device(d).type == "xpu", d will definitively not be "cpu"

torch.device(d).type == "xpu" is enough to check xpu device just like the case in MLU and NPU.

Who can review?

@SunMarc and @muellerzr

[faaany]

@abhilash1910

[muellerzr]

Thanks! Love the simplification 😍

Could we actually take this a bit further now that the logic is aligned?

Perhaps getting the device type expected under the is_xxx_available() and then only calling the

num_devices = len([d for d in max_memory if torch.device(d).type == expected_device_type and max_memory[d] > 0])

section once? WDYT

[abhilash1910]

Yes @muellerzr looks like a good abstraction, We are internally validating the "gpu" predicate of the validation logic .
If it is no longer required then new design LGTM.

[SunMarc]

LGTM !

[faaany]

@muellerzr PR updated with expected_device_type, pls review, thx!

[HuggingFaceDocBuilderDev]

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

[abhilash1910]

LGTM

[muellerzr]

Beautiful 😍

Nice work!