File tree 1 file changed +9
-1
lines changed
torch/testing/_internal/distributed 1 file changed +9
-1
lines changed Original file line number Diff line number Diff line change @@ -4807,7 +4807,11 @@ def _test_ddp_apply_optim_in_backward(
4807
4807
# set_to_none for regular optimizer to match in backward
4808
4808
# case.
4809
4809
optim .zero_grad (set_to_none = True )
4810
-
4810
+
4811
+ @skip_but_pass_in_sandcastle_if (
4812
+ BACKEND == "gloo" and HAS_TORCHVISION ,
4813
+ "Failing with gloo backend + torchvision due to ongoing issue https://github.com/pytorch/pytorch/issues/111834" ,
4814
+ )
4811
4815
@skip_if_lt_x_gpu (2 )
4812
4816
def test_ddp_apply_optim_in_backward (self ):
4813
4817
for optim_cls , init_before in itertools .product (
@@ -4820,6 +4824,10 @@ def test_ddp_apply_optim_in_backward(self):
4820
4824
init_before = init_before ,
4821
4825
)
4822
4826
4827
+ @skip_but_pass_in_sandcastle_if (
4828
+ BACKEND == "gloo" and HAS_TORCHVISION ,
4829
+ "Failing with gloo backend + torchvision due to ongoing issue https://github.com/pytorch/pytorch/issues/111834" ,
4830
+ )
4823
4831
@skip_if_lt_x_gpu (2 )
4824
4832
def test_ddp_apply_optim_in_backward_grad_as_bucket_view_false (self ):
4825
4833
for init_before in [True , False ]:
You can’t perform that action at this time.
0 commit comments