frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Revert "[2/N]Port several test files under test/distributed to Intel GPU (#159473)"

Browse Source 
This reverts commit 040d00af048967dde7938d358d7f5988cbd18388.

Reverted https://github.com/pytorch/pytorch/pull/159473 on behalf of https://github.com/jeanschmidt due to Seems to be breaking internal signals, @d4l3k please help the author to have this change landed. [D81718444](https://www.internalfb.com/diff/D81718444) ([comment](https://github.com/pytorch/pytorch/pull/159473#issuecomment-3264046983))
This commit is contained in:
PyTorch MergeBot
2025-09-07 21:06:37 +00:00
parent 8235c4f65d
commit ff2de5d522
11 changed files with 234 additions and 340 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
torch/distributed/distributed_c10d.py
View File

@ -337,12 +337,10 @@ class Backend(str): # noqa: SLOT000
# assume default devices "cpu" and "cuda", but warn
warnings.warn(
f"Device capability of {name} unspecified, assuming `cpu` and "
"`cuda` or `xpu`. Please specify it via the `devices` argument of "
"`cuda`. Please specify it via the `devices` argument of "
"`register_backend`."
)
Backend.backend_capability[name.lower()] = (
["cpu", "cuda", "xpu"] if torch.xpu.is_available() else ["cpu", "cuda"]
)
Backend.backend_capability[name.lower()] = ["cpu", "cuda"]
elif isinstance(devices, str):
# Single device string specified. Simply convert to list.
Backend.backend_capability[name.lower()] = [devices]