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

10
torch/testing/_internal/common_distributed.py
View File

@ -349,7 +349,15 @@ def requires_nccl_version(version, msg):
f"Requires NCCL version greater than or equal to: {version}, found: {torch.cuda.nccl.version()}, reason: {msg}",
)
else:
return lambda f: f
def decorator(func):
@wraps(func)
def wrapper(*args, **kwargs):
return func(*args, **kwargs)
return wrapper
return decorator
def requires_nccl():