pytorch

mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00

Files

Yu, Guangye e32a16a9da Correct torch.xpu.is_bf16_supported return False if no XPU detected (#152317 )

# Motivation
Fix https://github.com/pytorch/pytorch/issues/152301
When XPU is not available, calling `torch.xpu.is_bf16_supported()` still returns `True`, which is inconsistent with the expected behavior (should be False).

# Solution
Align to other backend, adding `including_emulation` to `torch.xpu.is_bf16_supported` and,
- return `False` if XPU is not available
- return `True` if `including_emulation` is True
- return `torch.xpu.get_device_properties().has_bfloat16_conversions` if `including_emulation` is False, it means if the device could generate SPIRV code for bf16.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/152317
Approved by: https://github.com/EikanWang

2025-05-06 10:03:17 +00:00

__init__.py

Correct torch.xpu.is_bf16_supported return False if no XPU detected (#152317 )

2025-05-06 10:03:17 +00:00

_gpu_trace.py

[BE][Easy] enable PYFMT for torch/[a-s]*/ (#138447 )

2024-12-23 14:04:00 +00:00

_utils.py

refactor code to share across different devices (#120602 )

2024-02-28 09:42:58 +00:00

memory.py

PEP585 update - torch/nn torch/optim torch/package torch/profiler torch/serialization torch/sparse torch/xpu (#145175 )