File tree Expand file tree Collapse file tree 1 file changed +9
-1
lines changed
torch/testing/_internal/distributed Expand file tree Collapse file tree 1 file changed +9
-1
lines changed Original file line number Diff line number Diff line change @@ -5066,7 +5066,11 @@ def _test_ddp_apply_optim_in_backward(
5066
5066
# set_to_none for regular optimizer to match in backward
5067
5067
# case.
5068
5068
optim .zero_grad (set_to_none = True )
5069
-
5069
+
5070
+ @skip_but_pass_in_sandcastle_if (
5071
+ BACKEND == "gloo" and HAS_TORCHVISION ,
5072
+ "Failing with gloo backend + torchvision due to ongoing issue https://github.com/pytorch/pytorch/issues/111834" ,
5073
+ )
5070
5074
@skip_if_lt_x_gpu (2 )
5071
5075
def test_ddp_apply_optim_in_backward (self ):
5072
5076
for optim_cls , init_before in itertools .product (
@@ -5079,6 +5083,10 @@ def test_ddp_apply_optim_in_backward(self):
5079
5083
init_before = init_before ,
5080
5084
)
5081
5085
5086
+ @skip_but_pass_in_sandcastle_if (
5087
+ BACKEND == "gloo" and HAS_TORCHVISION ,
5088
+ "Failing with gloo backend + torchvision due to ongoing issue https://github.com/pytorch/pytorch/issues/111834" ,
5089
+ )
5082
5090
@skip_if_lt_x_gpu (2 )
5083
5091
def test_ddp_apply_optim_in_backward_grad_as_bucket_view_false (self ):
5084
5092
for init_before in [True , False ]:
You can’t perform that action at this time.
0 commit comments