From 8221ee6db9fd319c249087ad9bb9abaa7d56e113 Mon Sep 17 00:00:00 2001 From: Frost Mitchell Date: Thu, 30 Oct 2025 19:29:06 +0000 Subject: [PATCH] [xpu] Fix type annotation for ProcessGroupXCCL (#166418) 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 --- torch/_C/_distributed_c10d.pyi | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/torch/_C/_distributed_c10d.pyi b/torch/_C/_distributed_c10d.pyi index da59123625e..737362be62b 100644 --- a/torch/_C/_distributed_c10d.pyi +++ b/torch/_C/_distributed_c10d.pyi @@ -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,