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,