OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[xpu] Fix type annotation for ProcessGroupXCCL (#166418)

Browse Source 
After #163049, this PR fixes the type annotations to match the actual implementation for ProcessGroupXCCL::Options.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/166418
Approved by: https://github.com/guangyey, https://github.com/ezyang
This commit is contained in:
Frost Mitchell 2025-10-30 19:29:06 +00:00 committed by PyTorch MergeBot
parent b939de26d1
commit 8221ee6db9
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
torch/_C/_distributed_c10d.pyi
View File

@ -849,7 +849,9 @@ class _SymmetricMemory:
class ProcessGroupXCCL(Backend):
class Options(Backend.Options):
def __init__(self): ...
is_high_priority_stream: bool
def __init__(self, is_high_priority_stream: bool = False): ...
def __init__(
self,