File tree Expand file tree Collapse file tree 1 file changed +1
-17
lines changed Expand file tree Collapse file tree 1 file changed +1
-17
lines changed Original file line number Diff line number Diff line change @@ -1101,23 +1101,7 @@ def __init__(self, manager: "Manager") -> None:
1101
1101
self ._manager = manager
1102
1102
1103
1103
def allreduce (self , tensors : List [torch .Tensor ], opts : object ) -> Work :
1104
- # Ensure we have a valid quorum and are configured before trying to do
1105
- # any work.
1106
- self ._manager .wait_quorum ()
1107
-
1108
- if self ._manager .errored () is not None :
1109
- return _DummyWork (tensors )
1110
- try :
1111
- work = super ().allreduce (tensors , opts )
1112
- except Exception as e :
1113
- self ._manager .report_error (e )
1114
- return _DummyWork (tensors )
1115
-
1116
- return _ManagedWork (
1117
- self ._manager ,
1118
- work ,
1119
- tensors ,
1120
- )
1104
+ return self ._manager .allreduce (tensors )
1121
1105
1122
1106
def size (self ) -> int :
1123
1107
return self ._manager .num_participants ()
You can’t perform that action at this time.
0 commit comments