Fix CI hopefully

Author: zou3519

test/test_ops.py

@@ -179,6 +179,8 @@ def is_inplace(op, variant):
 
 
 vjp_fail = {
+    xfail('nn.functional.dropout'),  # randomness testing artifact
+    xfail('nn.functional.rrelu'),  # randomness testing artifact
     xfail('linalg.cholesky'),
     xfail('linalg.inv'),
     xfail('linalg.matrix_power'),
@@ -234,6 +236,9 @@ def wrapped_fn(*args, **kwargs):
 
     @ops(functorch_lagging_op_db + additional_op_db, allowed_dtypes=(torch.float,))
     @skipOps('TestOperators', 'test_jvp', set({
+        xfail('nn.functional.dropout'),  # randomness testing artifact; not actually a problem
+        xfail('nn.functional.rrelu'),  # randomness testing artifact; not actually a problem
+
         # See https://github.com/pytorch/pytorch/issues/69034
         # RuntimeError: expected scalar type double but found float
         xfail('minimum'),

test/test_vmap.py

@@ -2873,7 +2873,6 @@ def test_div(self, device):
         self._test_arithmetic(lambda x, y: x / y, device)
 
     @allowVmapFallbackUsage
-    @unittest.expectedFailure
     def test_binary_cross_entropy(self, device):
         x = F.sigmoid(torch.randn(3, 2, device=device, requires_grad=True))
         target = torch.rand(3, 2, device=device)
@@ -3163,7 +3162,6 @@ def test_vmap_exhaustive(self, device, dtype, op):
         xfail('cdist'),
         xfail('complex'),
         xfail('copysign'),
-        xfail('diag_embed'),
         xfail('dsplit'),
         xfail('eig'),
         xfail('fft.fftn'),
@@ -3223,7 +3221,6 @@ def test_vmap_exhaustive(self, device, dtype, op):
         xfail('linalg.multi_dot'),
         xfail('nanmean'),
         xfail('vstack'),
-        xfail('block_diag'),
         xfail('nn.functional.dropout'),
         xfail('nn.functional.conv2d', ''),
         xfail('nn.functional.batch_norm'),