resolve_neg, resolve_conj

Author: zou3519

functorch/csrc/BatchRulesStopDecomposition.cpp

@@ -509,8 +509,6 @@ TORCH_LIBRARY_IMPL(aten, FT_BATCHED_KEY, m) {
   STOP_DECOMPOSE(repeat_interleave.self_Tensor);
   STOP_DECOMPOSE(repeat_interleave.self_int);
   STOP_DECOMPOSE(requires_grad_);
-  STOP_DECOMPOSE(resolve_conj);
-  STOP_DECOMPOSE(resolve_neg);
   STOP_DECOMPOSE(retain_grad);
   STOP_DECOMPOSE(retains_grad);
   STOP_DECOMPOSE(rnn_relu.data);

test/test_vmap.py

@@ -2990,8 +2990,6 @@ class TestVmapOperatorsOpInfo(TestCase):
         xfail('hsplit'),
         xfail('nn.functional.pad', 'circular'),
         xfail('resize_as_'),
-        xfail('resolve_conj'),
-        xfail('resolve_neg'),
         xfail('tensor_split'),
         xfail('to_sparse'),
         xfail('vsplit'),
@@ -3032,7 +3030,6 @@ def test_vmap_exhaustive(self, device, dtype, op):
 
     @ops(functorch_lagging_op_db + additional_op_db, allowed_dtypes=(torch.float,))
     @skipOps('TestVmapOperatorsOpInfo', 'test_op_has_batch_rule', {
-        # xfail('__getitem__'),
         xfail('cdist'),
         xfail('complex'),
         xfail('copysign'),
@@ -3087,8 +3084,7 @@ def test_vmap_exhaustive(self, device, dtype, op):
         xfail('renorm'),
         xfail('repeat_interleave'),
         xfail('resize_as_'),
-        xfail('resolve_conj'),
-        xfail('resolve_neg'),
+        xfail('scatter'),
         xfail('take'),
         xfail('take_along_dim'),
         xfail('tensor_split'),