fix CI

Author: Chillee

functorch/csrc/BatchRulesStopDecomposition.cpp

@@ -657,6 +657,8 @@ TORCH_LIBRARY_IMPL(aten, FT_BATCHED_KEY, m) {
   STOP_DECOMPOSE(linalg_inv);
   STOP_DECOMPOSE(linalg_matrix_power);
   STOP_DECOMPOSE(linalg_matrix_rank);
+  STOP_DECOMPOSE(linalg_matrix_rank.atol_rtol_tensor);
+  STOP_DECOMPOSE(linalg_matrix_rank.atol_rtol_float);
   STOP_DECOMPOSE(logical_and);
   STOP_DECOMPOSE(masked_select_backward);
   STOP_DECOMPOSE(matrix_exp_backward);

test/test_ops.py

@@ -426,10 +426,13 @@ def test_vmapvjp(self, device, dtype, op):
         xfail('linalg.eigh'),
         xfail('linalg.eigvals'),
         xfail('linalg.householder_product'),
+        xfail('linalg.lstsq'),
         xfail('linalg.inv'),
         xfail('linalg.matrix_norm'),
         xfail('linalg.matrix_power'),
         xfail('linalg.norm'),
+        xfail('linalg.pinv'),
+        xfail('linalg.pinv', 'hermitian'),
         xfail('linalg.slogdet'),
         xfail('linalg.solve'),
         xfail('linalg.tensorinv'),
@@ -467,6 +470,7 @@ def test_vmapvjp(self, device, dtype, op):
         xfail('norm', 'fro'),
         xfail('norm', 'inf'),
         xfail('norm', 'nuc'),
+        xfail('pinverse'),
         xfail('pow'),
         xfail('prod'),
         xfail('put'),

test/test_pythonkey.py

@@ -193,6 +193,7 @@ class TestPythonKeyOperatorsOpInfo(TestCase):
     xfail('linalg.matrix_power'),
     xfail('linalg.inv'),
     xfail('linalg.cholesky'),
+    xfail('nn.functional.dropout'),
     xfail('linalg.eigvals'),
     xfail('nn.functional.pad', 'circular'),
     })
@@ -293,22 +294,21 @@ class TestEagerFusionOpInfo(TestCase):
         xfail('matmul'),
         xfail('nn.functional.gelu'),
         xfail('nn.functional.linear'),
+        xfail('nn.functional.dropout'),
         xfail('polar'),
         xfail('special.zeta', 'grad'),
         xfail('to_sparse'),
         xfail('addcdiv'),
-        xfail('atanh'),
-        xfail('addcdiv'),
-        xfail('atanh'),
+        xfail('angle'),
         xfail('cholesky'),
         xfail('cumulative_trapezoid'),
         xfail('diag_embed'),
         xfail('linalg.householder_product'),
         xfail('logit'),
         xfail('matrix_exp'),
+        xfail('sgn'),
         xfail('trapezoid'),
         xfail('trapz'),
-        xfail('block_diag'),
     })
     def test_eager_compilation_exhaustive(self, device, dtype, op):
 

test/test_vmap.py

@@ -3066,6 +3066,8 @@ def test_vmap_exhaustive(self, device, dtype, op):
         xfail('linalg.matrix_power'),
         xfail('linalg.matrix_rank'),
         xfail('linalg.matrix_rank', 'hermitian'),
+        xfail('linalg.pinv'),
+        xfail('linalg.pinv', 'hermitian'),
         xfail('linalg.norm'),
         xfail('linalg.solve'),
         xfail('linalg.svdvals'),